diff --git a/src/comfy_pack/service.py b/src/comfy_pack/service.py
index 51d6253..4d49083 100644
--- a/src/comfy_pack/service.py
+++ b/src/comfy_pack/service.py
@@ -129,13 +129,6 @@ def generate(
             ret = ret[-1]
         return ret
 
-    @bentoml.on_shutdown
-    def on_shutdown(self):
-        logger.info("Shutting down")
-        if not EXISTING_COMFYUI_SERVER:
-            self.watch_thread.join()
-            logger.info("Watch thread finished")
-
     @bentoml.on_deployment
     @staticmethod
     def prepare_models():