diff --git a/src/serve.py b/src/serve.py
index bd4fece..84808f1 100644
--- a/src/serve.py
+++ b/src/serve.py
@@ -21,7 +21,7 @@ def main():
         return pipeline(pdf)
 
     prediction_server = make_prediction_server(predict)
-    run_prediction_server(prediction_server, CONFIG.webserver.host, CONFIG.webserver.port, CONFIG.webserver.mode)
+    run_prediction_server(prediction_server, CONFIG.webserver.host, CONFIG.webserver.port)
 
 
 if __name__ == "__main__":