diff --git a/code/e2e-tests/util.ts b/code/e2e-tests/util.ts
index bb017f54e12..bd15bfa1a56 100644
--- a/code/e2e-tests/util.ts
+++ b/code/e2e-tests/util.ts
@@ -87,14 +87,9 @@ export class SbPage {
 
   async waitForStoryLoaded() {
     try {
-      // wait for the story to be visitedz
+      // wait for the story to be visited
       await this.page.waitForURL((url) => url.search.includes(`path`));
 
-      // TODO: Remove this once docs are actually working
-      if (this.page.url().includes('/docs/')) {
-        return;
-      }
-
       const root = this.previewRoot();
       // Wait until there is at least one child (a story element) in the preview iframe
       await root.locator(':scope > *').first().waitFor({