From ef92a43608ca73dea810e37ea060d05794dab76e Mon Sep 17 00:00:00 2001
From: Yann Braga <yannbf@gmail.com>
Date: Wed, 8 Jan 2025 16:04:12 +0100
Subject: [PATCH] revert workaround

---
 code/e2e-tests/util.ts | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

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({