diff --git a/nix/pkgs/default/default.nix b/nix/pkgs/default/default.nix
index fc9ef5a7..543537ff 100644
--- a/nix/pkgs/default/default.nix
+++ b/nix/pkgs/default/default.nix
@@ -15,6 +15,9 @@
 , all-features ? false
 , features ? []
 , profile ? "release"
+, version-extra ? inputs.self.shortRev
+    or inputs.self.dirtyShortRev
+    or null,
 }:
 
 let
@@ -58,13 +61,8 @@ let
     });
 
   buildPackageEnv =
-    let
-      GRAPEVINE_VERSION_EXTRA = inputs.self.shortRev
-        or inputs.self.dirtyShortRev
-        or null;
-    in
-    (lib.optionalAttrs (GRAPEVINE_VERSION_EXTRA != null) {
-      inherit GRAPEVINE_VERSION_EXTRA;
+    (lib.optionalAttrs (version-extra != null) {
+      GRAPEVINE_VERSION_EXTRA = version-extra;
     })
     // buildDepsOnlyEnv;