diff --git a/src/types/comfy.d.ts b/src/types/comfy.d.ts
index c15fae9ff..0043da1b2 100644
--- a/src/types/comfy.d.ts
+++ b/src/types/comfy.d.ts
@@ -1,5 +1,5 @@
 import { LGraphNode, IWidget } from './litegraph'
-import { ComfyApp } from '../../scripts/app'
+import { ComfyApp } from '../scripts/app'
 
 export interface ComfyExtension {
   /**
diff --git a/tests-ui/utils/ezgraph.ts b/tests-ui/utils/ezgraph.ts
index 17d0ada52..a67a85aab 100644
--- a/tests-ui/utils/ezgraph.ts
+++ b/tests-ui/utils/ezgraph.ts
@@ -1,7 +1,7 @@
 import type { LiteGraph, LGraphCanvas } from '@comfyorg/litegraph'
 
 /**
- * @typedef { import("./src/scripts/app")["app"] } app
+ * @typedef { import("../../src/scripts/app")["app"] } app
  * @typedef { import("../../src/types/litegraph") } LG
  * @typedef { import("../../src/types/litegraph").IWidget } IWidget
  * @typedef { import("../../src/types/litegraph").ContextMenuItem } ContextMenuItem
diff --git a/tests-ui/utils/index.ts b/tests-ui/utils/index.ts
index 522cd3f3a..f8eb96123 100644
--- a/tests-ui/utils/index.ts
+++ b/tests-ui/utils/index.ts
@@ -67,7 +67,7 @@ export async function checkBeforeAndAfterReload(graph, cb) {
  * @param { string } name
  * @param { Record<string, string | [string | string[], any]> } input
  * @param { (string | string[])[] | Record<string, string | string[]> } output
- * @returns { Record<string, import("./src/types/comfy").ComfyObjectInfo> }
+ * @returns { Record<string, import("../../src/types/comfy").ComfyObjectInfo> }
  */
 export function makeNodeDef(name, input, output = {}) {
   const nodeDef = {
diff --git a/tests-ui/utils/setup.ts b/tests-ui/utils/setup.ts
index 05856cffc..808ff51ae 100644
--- a/tests-ui/utils/setup.ts
+++ b/tests-ui/utils/setup.ts
@@ -26,7 +26,7 @@ export interface APIConfig {
 }
 
 /**
- * @typedef { import("./src/types/comfy").ComfyObjectInfo } ComfyObjectInfo
+ * @typedef { import("../../src/types/comfy").ComfyObjectInfo } ComfyObjectInfo
  */
 
 /**