🪛

aslilac · aslilac · commit 0cdb2a1f18ff · 2024-02-26T20:25:50.000Z
diff --git a/site/src/pages/WorkspacePage/WorkspaceScheduleControls.test.tsx b/site/src/pages/WorkspacePage/WorkspaceScheduleControls.test.tsx
@@ -82,7 +82,7 @@ test("add 3 hours to deadline", async () => {
   await screen.findByText(
     "Workspace shutdown time has been successfully updated.",
   );
-  expect(screen.getByText("Stop in 6 hours")).toBeInTheDocument();
+  expect(await screen.findByText("Stop in 6 hours")).toBeInTheDocument();
 
   // Mocks are used here because the 'usedDeadline' is a dayjs object, which
   // can't be directly compared.
@@ -110,7 +110,7 @@ test("remove 2 hours to deadline", async () => {
   await screen.findByText(
     "Workspace shutdown time has been successfully updated.",
   );
-  expect(screen.getByText("Stop in an hour")).toBeInTheDocument();
+  expect(await screen.findByText("Stop in an hour")).toBeInTheDocument();
 
   // Mocks are used here because the 'usedDeadline' is a dayjs object, which
   // can't be directly compared.
