diff --git a/nix/homes/x86_64-linux/khs/default.nix b/nix/homes/x86_64-linux/khs/default.nix
index f6c613a..1fd2522 100644
--- a/nix/homes/x86_64-linux/khs/default.nix
+++ b/nix/homes/x86_64-linux/khs/default.nix
@@ -1,10 +1,8 @@
 {
-  lib,
-  config,
   ...
 }:
 {
   khscodes.khs.enable = true;
   khscodes.khs.shell.oh-my-posh.enable = true;
-  imports = lib.lists.optional config.khscodes.desktop.enable ./desktop.nix;
+  imports = [ ./desktop.nix ];
 }
diff --git a/nix/homes/x86_64-linux/khs/desktop.nix b/nix/homes/x86_64-linux/khs/desktop.nix
index ce11105..50c36bd 100644
--- a/nix/homes/x86_64-linux/khs/desktop.nix
+++ b/nix/homes/x86_64-linux/khs/desktop.nix
@@ -1,7 +1,12 @@
-{ pkgs, lib, ... }:
 {
-  imports = lib.lists.optional (lib.strings.hasSuffix "-linux" pkgs.system) ./linux-desktop.nix;
-  home.packages = [
+  pkgs,
+  lib,
+  config,
+  ...
+}:
+{
+  imports = [ ./linux-desktop.nix ];
+  home.packages = lib.mkIf config.khscodes.desktop.enable [
     pkgs.bitwarden-cli
     pkgs.nerd-fonts.inconsolata
     pkgs.google-chrome
diff --git a/nix/homes/x86_64-linux/khs/linux-desktop.nix b/nix/homes/x86_64-linux/khs/linux-desktop.nix
index dc457e5..9769a56 100644
--- a/nix/homes/x86_64-linux/khs/linux-desktop.nix
+++ b/nix/homes/x86_64-linux/khs/linux-desktop.nix
@@ -1,4 +1,9 @@
-{ pkgs, ... }:
 {
-  home.packages = [ pkgs.spotify ];
+  pkgs,
+  config,
+  lib,
+  ...
+}:
+{
+  home.packages = lib.mkIf config.khscodes.desktop.enable [ pkgs.spotify ];
 }