diff --git a/nixos/code.nix b/nixos/code.nix
index 462ae206..aaf118a9 100644
--- a/nixos/code.nix
+++ b/nixos/code.nix
@@ -6,6 +6,13 @@
   makeEnable,
   ...
 }:
+let
+  codexDesktop = inputs.codex-desktop-linux.packages.${pkgs.stdenv.hostPlatform.system}.default.overrideAttrs (oldAttrs: {
+    src = oldAttrs.src.overrideAttrs (_: {
+      outputHash = "sha256-FlPgkhlWpyjigv+r46srQlO7+zq8G3Ow6IMVpAUjCWQ=";
+    });
+  });
+in
 makeEnable config "myModules.code" true {
   programs.direnv = {
     enable = true;
@@ -60,7 +67,7 @@ makeEnable config "myModules.code" true {
       antigravity
       claude-code
       codex
-      inputs.codex-desktop-linux.packages.${pkgs.stdenv.hostPlatform.system}.default
+      codexDesktop
       gemini-cli
       happy-coder
       opencode
diff --git a/nixos/justfile b/nixos/justfile
index 405f3816..8978dc42 100644
--- a/nixos/justfile
+++ b/nixos/justfile
@@ -1,4 +1,5 @@
 switch *args:
+	if systemctl is-active --quiet nixos-upgrade.service; then sudo systemctl stop nixos-upgrade.service; fi
 	sudo nixos-rebuild switch --flake ".#" --impure {{args}}
 
 populate-cachix cache="colonelpanic8-dotfiles" host=`hostname` *args: