diff --git a/home-manager/common/image-specific.nix b/home-manager/common/image-specific.nix
deleted file mode 100644
index 9d06b066..00000000
--- a/home-manager/common/image-specific.nix
+++ /dev/null
@@ -1,4 +0,0 @@
-{ lib, ... }: {
-  # Requires secrets not available in images
-  programs.atuin.enable = lib.mkDefault false; 
-}
diff --git a/home-manager/image.nix b/home-manager/image.nix
index c30279ff..89f5ebf1 100644
--- a/home-manager/image.nix
+++ b/home-manager/image.nix
@@ -16,7 +16,6 @@
   ++ lib.optional (builtins.isString desktop) ./common/software/gui                    # GUI packages
   ++ lib.optional (builtins.isString desktop) ./common/desktops/${desktop}             # Machine-agnostic desktop configs
   ++ lib.optional (builtins.isString desktop) ./hosts/${hostname}/desktops/${desktop}  # Machine-specific desktop configs
-  ++ lib.optional (builtins.isString isImage) ./common/image-specific.nix              # Image specific items
   ;
 
   programs.atuin.enable = lib.mkDefault false; 
diff --git a/lib/default.nix b/lib/default.nix
index 211714e7..bd439d2a 100644
--- a/lib/default.nix
+++ b/lib/default.nix
@@ -168,9 +168,17 @@
     system    ? "x86_64-linux",
     gpu       ? null, 
     theme     ? "default",
+    repo      ? "nixpkgs",
+    unfree    ? false,
     format
   }: inputs.nixos-generators.nixosGenerate {
-    specialArgs = { inherit inputs outputs desktop hostname username stateVersion hmStateVersion gpu system theme format; };
+    specialArgs = { 
+      inherit inputs outputs desktop hostname username stateVersion hmStateVersion gpu system theme format;
+      # Choose whether to pull from stable or unstable 
+      pkgs          = let packages = (import ./packages.nix { inherit inputs repo system unfree; }); in packages.pkgs;
+      pkgs-unstable = let packages = (import ./packages.nix { inherit inputs repo system unfree; }); in packages.pkgs-unstable;
+    };
+
     system = system;
     format = format;
 
diff --git a/nixos/base.nix b/nixos/base.nix
index 24de6d15..dbf5d96e 100644
--- a/nixos/base.nix
+++ b/nixos/base.nix
@@ -1,10 +1,5 @@
 { inputs, lib,hostname, username, desktop, ... }: {
   imports = [ 
-    # Services
-    ./common/services/tailscale.nix
-    ./common/services/promtail.nix
-    ./common/services/telegraf.nix
-
     # NixOS Modules
     ./common/modules/networking.nix      # Initial Networking configs
     ./common/modules/nixos.nix           # Common NixOS Configurations
diff --git a/nixos/common/modules/installer.nix b/nixos/common/modules/installer.nix
index 87b7fc28..c3256da6 100644
--- a/nixos/common/modules/installer.nix
+++ b/nixos/common/modules/installer.nix
@@ -1,4 +1,4 @@
-{ config, desktop, lib, pkgs, username, ... }:
+{ pkgs, ... }:
 let
   install-system = pkgs.writeScriptBin "install-system" "${builtins.readFile ../../../docs/install.sh}";
 in
diff --git a/nixos/small.nix b/nixos/small.nix
index 49258063..0549b408 100644
--- a/nixos/small.nix
+++ b/nixos/small.nix
@@ -7,6 +7,10 @@
     # Services
     ./common/services/openssh.nix
     ./common/services/fail2ban.nix
+    ./common/services/tailscale.nix
+    ./common/services/promtail.nix
+    ./common/services/telegraf.nix
+
 
     # Software
     ./common/software/cli/scripts.nix