diff --git a/system-modules/ollama.nix b/system-modules/ollama.nix
index 3927290..c642bc8 100644
--- a/system-modules/ollama.nix
+++ b/system-modules/ollama.nix
@@ -1,17 +1,6 @@
-{ pkgs, pkgs-unstable, pkgs-latest, lib, host, nvidia, ... }: 
+{ pkgs, lib, host, nvidia, ... }: 
 let
-  p = pkgs-latest.ollama;
-  # p = pkgs-unstable.ollama.overrideAttrs (oldAttrs: rec {
-  #   inherit (oldAttrs) pname;
-  #   version = "0.3.0";
-  #   src = pkgs.fetchFromGitHub {
-  #     owner = "ollama";
-  #     repo = "ollama";
-  #     rev = "v${version}";
-  #     hash = "sha256-69CpRAggx6a1NJq+CA9QliXuUbDgC1ERRuA3y17KVAM=";
-  #     fetchSubmodules = true;
-  #   };   
-  # });
+  p = if nvidia.enable then pkgs.ollama-cuda else pkgs.ollama;
 in {
   environment.systemPackages = [ p ];