diff --git a/flake.nix b/flake.nix
index d690ab0..21528a0 100644
--- a/flake.nix
+++ b/flake.nix
@@ -3,8 +3,10 @@
   inputs = {
     nixpkgs.url = "github:NixOS/nixpkgs";
     flake-utils.url = "github:numtide/flake-utils";
-    # Cannot override nixpkgs input as nix-unit fails to build with nixVersions.git
-    nix-unit.url = "github:nix-community/nix-unit";
+    nix-unit = {
+      url = "github:nix-community/nix-unit";
+      inputs.nixpkgs.follows = "nixpkgs";
+    };
   };
   outputs =
     {