diff --git a/__visual_tests__/utils.ts b/__visual_tests__/utils.ts
index 835593151..547c320b1 100644
--- a/__visual_tests__/utils.ts
+++ b/__visual_tests__/utils.ts
@@ -16,9 +16,9 @@ export async function assertScreenshot(page: Page, screenshotName: string, brows
 export async function loadWidget(page: Page) {
   await page.click(WidgetComponentIds.WIDGET_BUTTON);
   // NOTE: below fails sometimes in CI.
-  // const widgetWindow = page.locator(WidgetComponentIds.MESSAGE_INPUT);
-  // await widgetWindow.waitFor({ state: 'visible' });
-  await page.waitForTimeout(3000);
+  const widgetWindow = page.locator(WidgetComponentIds.SUGGESTED_REPLIES_OPTIONS);
+  await widgetWindow.waitFor({ state: 'visible' });
+  // await page.waitForTimeout(3000);
 }
 
 export async function sendTextMessage(page: Page, text: string, waitTime = 1000) {