diff --git a/hosts/architect/llm.nix b/hosts/architect/llm.nix
index a1a8971..0bbae6a 100644
--- a/hosts/architect/llm.nix
+++ b/hosts/architect/llm.nix
@@ -4,6 +4,7 @@ let
   backendDomain = "ollama.giugl.io";
   frontendDomain = "llm.giugl.io";
   ollamaPkg = pkgs.unstablePkgs.ollama-cuda;
+  uiPkg = pkgs.unstablePkgs.open-webui;
 in
 {
   environment = {
@@ -23,7 +24,10 @@ in
       };
     };
 
-    open-webui.enable = true;
+    open-webui = {
+      enable = true;
+      package = uiPkg;
+    };
   };
 
   architect.vhost.${backendDomain} = {
@@ -51,6 +55,7 @@ in
       port = config.services.open-webui.port;
       allowLan = true;
       allowWAN = true;
+        proxyWebsockets = true;
     };
   };
 }