diff --git a/hosts/main/default.nix b/hosts/main/default.nix
index 35d3053..75d7df7 100644
--- a/hosts/main/default.nix
+++ b/hosts/main/default.nix
@@ -76,6 +76,7 @@
       enable = true;
       port = 1434;
       acceleration = "cuda";
+      package = pkgs.unstable.ollama;
       host = "0.0.0.0";
     };
     environment.sessionVariables = {