diff --git a/api/onnx_web/convert/diffusion/diffusers.py b/api/onnx_web/convert/diffusion/diffusers.py
index c4893fc1..1f4459c9 100644
--- a/api/onnx_web/convert/diffusion/diffusers.py
+++ b/api/onnx_web/convert/diffusion/diffusers.py
@@ -267,7 +267,7 @@ def convert_diffusion_diffusers(
                 "sample": {0: "batch", 1: "channels", 2: "height", 3: "width"},
             },
             opset=ctx.opset,
-            half=False, # https://github.com/ssube/onnx-web/issues/290
+            half=False,  # https://github.com/ssube/onnx-web/issues/290
         )
 
         # VAE DECODER
diff --git a/api/onnx_web/main.py b/api/onnx_web/main.py
index 21d95533..8cab4114 100644
--- a/api/onnx_web/main.py
+++ b/api/onnx_web/main.py
@@ -1,5 +1,6 @@
 import atexit
 import gc
+from functools import partial
 from logging import getLogger
 
 from diffusers.utils.logging import disable_progress_bar
@@ -67,11 +68,11 @@ def run():
     app, pool = main()
     pool.start()
 
-    def quit():
+    def quit(p: DevicePoolExecutor):
         logger.info("shutting down workers")
-        pool.join()
+        p.join()
 
-    atexit.register(quit)
+    atexit.register(partial(quit, pool))
     return app