diff --git a/js/common.js b/js/common.js
index d51485da..e393f0bd 100644
--- a/js/common.js
+++ b/js/common.js
@@ -1,8 +1,15 @@
 import { app } from "../../scripts/app.js";
 import { api } from "../../scripts/api.js";
-import { $el, ComfyDialog } from "../../scripts/ui.js";
 import { getBestPosition, getPositionStyle, getRect } from './popover-helper.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"));
+}
+
 
 function internalCustomConfirm(message, confirmMessage, cancelMessage) {
 	return new Promise((resolve) => {
@@ -667,4 +674,4 @@ function initTooltip () {
     document.body.addEventListener('mouseleave', mouseleaveHandler, true);
 }
 
-initTooltip();
\ No newline at end of file
+initTooltip();