diff --git a/flake.nix b/flake.nix
index 2713963..a6cd396 100644
--- a/flake.nix
+++ b/flake.nix
@@ -272,8 +272,8 @@
               useGlobalPkgs = true; # Use system packages globally
               useUserPackages = true; # Install packages to user profile
               extraSpecialArgs = {
-                stable = inputs.nixpkgs-s.legacyPackages.${config.nixpkgs.system}; # Pass stable channel to home-manager
-                unstable = inputs.nixpkgs-u.legacyPackages.${config.nixpkgs.system}; # Pass unstable channel to home-manager
+                stable = import inputs.nixpkgs-s { inherit (config.nixpkgs) system; }; # Pass stable channel to home-manager
+                unstable = import inputs.nixpkgs-u { inherit (config.nixpkgs) system; }; # Pass unstable channel to home-manager
               };
             };
           };
diff --git a/instances/thor/configuration.nix b/instances/thor/configuration.nix
index b5f0a51..3d2c639 100644
--- a/instances/thor/configuration.nix
+++ b/instances/thor/configuration.nix
@@ -1,6 +1,7 @@
 { lib
 , config
 , pkgs
+, unstable
 , ...
 }:
 {
@@ -213,6 +214,7 @@
     };
     # joycond.enable = true;
     keybase.enable = true;
+    ollama.package = unstable.ollama-vulkan;
     openssh = {
       enable = true;
       openFirewall = true;