diff --git a/system/services/containers/ollama/default.nix b/system/services/containers/ollama/default.nix
index 7755097..3843cff 100644
--- a/system/services/containers/ollama/default.nix
+++ b/system/services/containers/ollama/default.nix
@@ -72,7 +72,7 @@
                             system = "x86_64-linux";
                             config.allowUnfree = true;
                         }; 
-                    in pkgs-us.ollama;
+                    in pkgs-us.ollama-cuda;
 
                     host = "0.0.0.0";