fix: allow for task dataset download on new domain

NEUROFUSIONInc · May 22, 2024 · b791545 · b791545
1 parent eef7d1f
commit b791545
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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;