1 files changed, 8 insertions, 6 deletions
diff --git a/src/tests/pages/CallbackPage.test.tsx b/src/tests/pages/CallbackPage.test.tsx
index 70f2fed..37fb932 100644
--- a/src/tests/pages/CallbackPage.test.tsx
+++ b/src/tests/pages/CallbackPage.test.tsx
@@ -1,9 +1,9 @@
 import React from "react";
-import { render } from "@testing-library/react";
+import { render, waitFor } from "@testing-library/react";
 
 import CallbackPage from "../../pages/CallbackPage";
 
-test("callback page sends provided code", () => {
+test("callback page sends provided code", async () => {
     global.opener = {
         postMessage: jest.fn()
     };
@@ -14,9 +14,11 @@ test("callback page sends provided code", () => {
 
     render(<CallbackPage/>);
 
-    expect(global.opener.postMessage).toBeCalledTimes(1);
-    expect(global.opener.postMessage).toBeCalledWith({
-        code: "abcde_code",
-        state: "abcde_state"
+    await waitFor(() => {
+        expect(global.opener.postMessage).toBeCalledTimes(1);
+        expect(global.opener.postMessage).toBeCalledWith({
+            code: "abcde_code",
+            state: "abcde_state"
+        });
     });
 });