diff --git a/pkgs/by-name/az/az-pim-cli/package.nix b/pkgs/by-name/az/az-pim-cli/package.nix
index 8878135ed3e8..e595583cde3b 100644
--- a/pkgs/by-name/az/az-pim-cli/package.nix
+++ b/pkgs/by-name/az/az-pim-cli/package.nix
@@ -5,8 +5,9 @@
   installShellFiles,
   stdenv,
   buildPackages,
-  versionCheckHook,
   nix-update-script,
+  testers,
+  az-pim-cli,
 }:
 buildGoModule (finalAttrs: {
   pname = "az-pim-cli";
@@ -37,11 +38,13 @@ buildGoModule (finalAttrs: {
     ''
   );
 
-  doInstallCheck = true;
-  nativeInstallCheck = [ versionCheckHook ];
-  versionCheckProgramArg = "version";
-
-  passthru.updateScript = nix-update-script { };
+  passthru = {
+    updateScript = nix-update-script { };
+    tests.version = testers.testVersion {
+      command = "HOME=$TMPDIR az-pim-cli --version";
+      package = az-pim-cli;
+    };
+  };
 
   meta = {
     description = "List and activate Azure Entra ID Privileged Identity Management roles from the CLI";