diff --git a/pkgs/by-name/ba/balena-cli/package.nix b/pkgs/by-name/ba/balena-cli/package.nix
index da3b64c558f3..666b8c5f5f81 100644
--- a/pkgs/by-name/ba/balena-cli/package.nix
+++ b/pkgs/by-name/ba/balena-cli/package.nix
@@ -37,7 +37,6 @@ buildNpmPackage' rec {
   nativeBuildInputs = [
     node-gyp'
     python3
-    versionCheckHook
   ]
   ++ lib.optionals stdenv.hostPlatform.isDarwin [
     cctools
@@ -47,6 +46,9 @@ buildNpmPackage' rec {
     udev
   ];
 
+  nativeInstallCheckInputs = [
+    versionCheckHook
+  ];
   doInstallCheck = true;
   versionCheckProgram = "${placeholder "out"}/bin/balena";