diff --git a/pkgs/by-name/di/distribution/package.nix b/pkgs/by-name/di/distribution/package.nix
index 9ade1080e3e2..9da917b51218 100644
--- a/pkgs/by-name/di/distribution/package.nix
+++ b/pkgs/by-name/di/distribution/package.nix
@@ -2,6 +2,7 @@
 , buildGoModule
 , fetchFromGitHub
 , fetchpatch
+, nix-update-script
 , testers
 , distribution
 }:
@@ -45,6 +46,7 @@ buildGoModule rec {
       package = distribution;
       version = "v${version}";
     };
+    updateScript = nix-update-script { };
   };
 
   meta = with lib; {