diff --git a/js/comfyui-share-copus.js b/js/comfyui-share-copus.js
index 46288e59..cc0b933c 100644
--- a/js/comfyui-share-copus.js
+++ b/js/comfyui-share-copus.js
@@ -1,7 +1,14 @@
 import { app } from "../../scripts/app.js";
-import { $el, ComfyDialog } from "../../scripts/ui.js";
 import { customAlert } from "./common.js";
 
+// === SHIM FOR NEW COMFYUI (removes ui.js warning) ===
+let $el, ComfyDialog;
+if (window?.comfyAPI?.ui) {
+    ({ $el, ComfyDialog } = window.comfyAPI.ui);
+} else {
+    ({ $el, ComfyDialog } = await import("../../scripts/ui.js"));
+}
+
 const env = "prod";
 
 let DEFAULT_HOMEPAGE_URL = "https://copus.io";