diff --git a/flake.nix b/flake.nix
index b728840..ba0a5fb 100644
--- a/flake.nix
+++ b/flake.nix
@@ -20,7 +20,8 @@
     ];
   in
   {
-    packages.x86_64-linux.default = pkgs.python3Packages.buildPythonPackage rec {
+  
+  packages.x86_64-linux.default = pkgs.python3Packages.buildPythonPackage rec {
       pname = "ozai-webui";
       version = "0.1.1";
       #propagatedBuildInputs = deps;
@@ -37,6 +38,7 @@
         mkdir -p $out/lib/python3.11/site-packages/static
         mkdir -p $out/lib/python3.11/templates
         mkdir -p $out/lib/python3.11/static
+
         install -Dm444 ${src}/static/* $out/share/ozai_webui/static/
         install -Dm444 ${src}/templates/* $out/share/ozai_webui/templates/
         cp -r $src/templates $out/lib/python3.11/site-packages/ozai_webui/templates
@@ -122,18 +124,19 @@ packages.x86_64-linux.ozai-webui-run = pkgs.stdenv.mkDerivation rec {
           description = "Ozai WebUI server";
           after = [ "network.target" ];
           wantedBy = [ "multi-user.target" ];
-	  environment = {
-              OZAI_URL=  "${config.services.ozai-webui.ozaiUrl}";
-              OZAI_WEBUI_HOST= "${config.services.ozai-webui.host}";
-              OZAI_WEBUI_PORT= toString(config.services.ozai-webui.port);
-              OZAI_WEBUI_STATIC_FOLDER= "${self.packages.x86_64-linux.default}/share/ozai-webui/static";
-              OZAI_WEBUI_TEMPLATE_FOLDER="${self.packages.x86_64-linux.default}/share/ozai-webui/templates";
-            };
+          environment = {
+                    OZAI_URL=  "${config.services.ozai-webui.ozaiUrl}";
+                    OZAI_WEBUI_HOST= "${config.services.ozai-webui.host}";
+                    OZAI_WEBUI_PORT= toString(config.services.ozai-webui.port);
+                    OZAI_WEBUI_STATIC_FOLDER= "${self.packages.x86_64-linux.default}/share/ozai-webui/static";
+                    OZAI_WEBUI_TEMPLATE_FOLDER="${self.packages.x86_64-linux.default}/share/ozai-webui/templates";
+                  };
 
           serviceConfig = {
             #ExecStart = "${self.packages.x86_64-linux.ozai-webui-run}/bin/run ${config.services.ozai-webui.host} ${toString(config.services.ozai-webui.port)} ${self.packages.x86_64-linux.default}/bin";
-            ExecStart = "${self.packages.x86_64-linux.ozai-webui-run}/bin/run ${config.services.ozai-webui.host} ${toString(config.services.ozai-webui.port)} ${self.packages.x86_64-linux.default}/lib/python3.11/site-packages/";
-            #ExecStart = "${self.packages.x86_64-linux.default}/bin/ozai_webui";
+            # ExecStart = "${self.packages.x86_64-linux.ozai-webui-run}/bin/run ${config.services.ozai-webui.host} ${toString(config.services.ozai-webui.port)} ${self.packages.x86_64-linux.default}/lib/python3.11/site-packages/";
+            # ExecStart = "${self.packages.x86_64-linux.ozai-webui}/bin/ozai-webui";
+            ExecStart = "${self.packages.x86_64-linux.default}/bin/ozai_webui";
             Restart = "always";
           };
         };
diff --git a/result b/result
index 0ea6ef9..d487899 120000
--- a/result
+++ b/result
@@ -1 +1 @@
-/nix/store/22y4lw54f6yp05f6h0876m4rwfk7hi0k-ozai-webui-run-0.1.1
\ No newline at end of file
+/nix/store/v6rmds7152d3cg3h77kjsg1kfvmgz108-ozai-webui-run-0.1.1
\ No newline at end of file
diff --git a/result-1 b/result-1
deleted file mode 120000
index 3c45dd4..0000000
--- a/result-1
+++ /dev/null
@@ -1 +0,0 @@
-/nix/store/f0rv69ig250np27kqx6ag8sbyx5f05fy-python3.11-ozai-webui-0.1.1
\ No newline at end of file
diff --git a/src/ozai_webui/__init__.py b/src/ozai_webui/__init__.py
index ccc557f..e6cab9d 100755
--- a/src/ozai_webui/__init__.py
+++ b/src/ozai_webui/__init__.py
@@ -221,8 +221,9 @@ def ws_message(data):
         socketio.emit('move_status', response.text, room=game_id)
 
 def main ():
-    return app.run(host=ozai_webui_host, port=ozai_webui_port)
+    # return app.run(host=ozai_webui_host, port=ozai_webui_port)
     #return socketio.run(app, debug=False, host=ozai_webui_host, port=ozai_webui_port, allow_unsafe_werkzeug=True)
+    return socketio.run(app, debug=False, host=ozai_webui_host, port=ozai_webui_port)
 
 if __name__ == '__main__':
     main()