diff --git a/src/frontend/tests/extended/integrations/chatInputOutputUser-shard-1.spec.ts b/src/frontend/tests/extended/integrations/chatInputOutputUser-shard-1.spec.ts
index f0715cfd4..3118d6bb1 100644
--- a/src/frontend/tests/extended/integrations/chatInputOutputUser-shard-1.spec.ts
+++ b/src/frontend/tests/extended/integrations/chatInputOutputUser-shard-1.spec.ts
@@ -172,7 +172,7 @@ test(
     await page.waitForTimeout(600);
     await page.keyboard.press("o");
     await page.waitForSelector(
-      `[data-testid="${urlNodeId}-data-output-modal"]`,
+      `[data-testid="${urlNodeId}-dataframe-output-modal"]`,
       {
         timeout: 3000,
       },