diff --git a/jest.setup.ts b/jest.setup.ts
index 19f6a05..4b4efde 100644
--- a/jest.setup.ts
+++ b/jest.setup.ts
@@ -9,9 +9,12 @@ if (process.env.NODE_ENV === 'production') {
   // Override React.act for production builds
   const originalAct = React.act;
   if (!originalAct) {
-    React.act = (callback: () => void) => {
-      callback();
-    };
+    React.act = (callback: () => void | Promise<void>) => {
+      const result = callback();
+      if (result instanceof Promise) {
+        return result;
+      }
+    } as typeof React.act;
   }
   
   // Also mock the act function from react-dom/test-utils