diff --git a/modules/llama-swap.nix b/modules/llama-swap.nix
index 0ff49d6..b93bc2c 100644
--- a/modules/llama-swap.nix
+++ b/modules/llama-swap.nix
@@ -2,6 +2,8 @@
   config,
   pkgs,
   lib,
+  inputs,
+  system,
   ...
 }:
 {
@@ -19,8 +21,8 @@
 
   services.llama-swap = {
     enable = true;
-    package = pkgs.unstable.llama-swap;
-    port = 11111;
+    package = inputs.self.packages.${system}.llama-swap;
+    port = 11112;
     openFirewall = true;
 
     settings =