diff --git a/api/onnx_web/main.py b/api/onnx_web/main.py
index 54f2f7df..d1b20a14 100644
--- a/api/onnx_web/main.py
+++ b/api/onnx_web/main.py
@@ -69,6 +69,7 @@ def run():
 
 
 if __name__ == "__main__":
-    app = run()
+    app, pool = main()
     app.run("0.0.0.0", 5000, debug=is_debug())
     logger.info("shutting down app")
+    pool.join()