diff --git a/desktop/developer.nix b/desktop/developer.nix
index df230b1..e82ac2b 100644
--- a/desktop/developer.nix
+++ b/desktop/developer.nix
@@ -1,4 +1,4 @@
-{ config, pkgs, lib, nixpkgs-unstable, ... }:
+{ config, pkgs, lib, ... }:
 with lib;
 let
   cfg = config.aviallon.developer;
@@ -131,10 +131,10 @@ in {
       user = "ollama";
       package =
         if config.aviallon.hardware.amd.enable
-          then (pkgs.callPackage (nixpkgs-unstable + "pkgs/by-name/ol/ollama") { acceleration = "rocm"; })
+          then pkgs.unstable.ollama-rocm
         else if (config.aviallon.hardware.nvidia.enable && config.aviallon.hardware.nvidia.variant != "nouveau")
-          then (pkgs.callPackage (nixpkgs-unstable + "pkgs/by-name/ol/ollama") { acceleration = "cuda"; })
-        else (pkgs.callPackage (nixpkgs-unstable + "pkgs/by-name/ol/ollama") {})
+          then pkgs.unstable.ollama-cuda
+        else pkgs.unstable.ollama
       ;
     };