diff --git a/frontend/src/components/lab/experiment.tsx b/frontend/src/components/lab/experiment.tsx
index 0aaef4d5..1d179ebb 100644
--- a/frontend/src/components/lab/experiment.tsx
+++ b/frontend/src/components/lab/experiment.tsx
@@ -100,7 +100,7 @@ export const Experiment: FC<IExperiment> = (experiment) => {
     window.addEventListener("message", (event) => {
       // IMPORTANT: Check the origin of the data!
       // You should probably not use '*', but restrict it to certain domains:
-      if (event.origin.startsWith("https://localhost:") || event.origin.startsWith("https://usefusion.app")) {
+      if (event.origin.startsWith("https://localhost:") || event.origin.startsWith("https://usefusion.app") || event.origin.startsWith("https://usefusion.ai")) {
         console.log("event", event);
         if (typeof event.data === "string") {
           return;