diff --git a/server/utils/extendChain.ts b/server/utils/extendChain.ts
index 1501602..5d2cada 100644
--- a/server/utils/extendChain.ts
+++ b/server/utils/extendChain.ts
@@ -16,7 +16,8 @@ import {
   removeExtendTemp
 } from '~/server/utils/library'
 import { extractLastFrame, probeHasAudio } from '~/server/utils/ffmpeg'
-import { ensureComfyReady, waitForComfySocket, watchComfyJob } from '~/server/utils/comfyLifecycle'
+import { ensureComfyReady } from '~/server/utils/comfyLifecycle'
+import { waitForComfySocket, watchComfyJob } from '~/server/utils/watch'
 import { comfyFilenamePrefix, queuePrompt, uploadImage } from '~/server/utils/comfy'
 import { buildWorkflow } from '~/server/utils/workflow'
 import { writePendingJob } from '~/server/utils/pending'