diff --git a/nixos/flake.lock b/nixos/flake.lock
index 8a3e8743..4a169d08 100644
--- a/nixos/flake.lock
+++ b/nixos/flake.lock
@@ -158,16 +158,16 @@
         ]
       },
       "locked": {
-        "lastModified": 1783616128,
-        "narHash": "sha256-GPBw1CWo9ZNDNDXgjw0Ar7gPFuZlloDzuOjEk6q7ADc=",
-        "owner": "sadjow",
+        "lastModified": 1783622783,
+        "narHash": "sha256-MVNYVjkYcoJfRruUNegFilJjFn15K+7sOo3noM2IC8A=",
+        "owner": "aroman",
         "repo": "codex-cli-nix",
-        "rev": "be567a404cddca0c018b391b11e179ae968e7a27",
+        "rev": "1e7d9d1751717416730dadceb97b222f4a5bf047",
         "type": "github"
       },
       "original": {
-        "owner": "sadjow",
-        "ref": "main",
+        "owner": "aroman",
+        "ref": "native-code-mode-host",
         "repo": "codex-cli-nix",
         "type": "github"
       }
diff --git a/nixos/flake.nix b/nixos/flake.nix
index e228fea1..e859c217 100644
--- a/nixos/flake.nix
+++ b/nixos/flake.nix
@@ -275,8 +275,8 @@
     nixtheplanet.url = "github:matthewcroughan/nixtheplanet";
 
     codex-cli-nix = {
-      # Default branch is `main` on GitHub (not `master`).
-      url = "github:sadjow/codex-cli-nix/main";
+      # TODO: revert to sadjow/codex-cli-nix main once PR #122 (code-mode-host) merges
+      url = "github:aroman/codex-cli-nix/native-code-mode-host";
       inputs = {
         nixpkgs.follows = "nixpkgs";
         flake-utils.follows = "flake-utils";