diff --git a/flaskr/app.py b/flaskr/app.py
index eb625a0..32a4f30 100644
--- a/flaskr/app.py
+++ b/flaskr/app.py
@@ -5,11 +5,11 @@ from flask_socketio import SocketIO, emit
 import logging
 
 
-def create_flask_app(s2c, c2s, s2flask, args):
+def create_flask_app(s2c, c2s, s2flask, kwargs):
     app = Flask(__name__, template_folder="templates", static_folder="static")
     log = logging.getLogger('werkzeug')
     log.disabled = True
-    model_path = Path(args.get('model_path', ''))
+    model_path = Path(kwargs.get('saved_models_path', ''))
     filename = 'preview.jpg'
     preview_file = str(model_path / filename)