diff --git a/pkgs/servers/sql/postgresql/postgresqlBuildExtension.nix b/pkgs/servers/sql/postgresql/postgresqlBuildExtension.nix
index 9ea82fb80f84..e328a0783993 100644
--- a/pkgs/servers/sql/postgresql/postgresqlBuildExtension.nix
+++ b/pkgs/servers/sql/postgresql/postgresqlBuildExtension.nix
@@ -63,10 +63,14 @@
   nix-update-script,
 }:
 
-args:
+lib.extendMkDerivation {
+  constructDrv = stdenv.mkDerivation;
 
-let
-  postgresqlBuildExtension =
+  excludeDrvArgNames = [
+    "enableUpdateScript"
+  ];
+
+  extendDrvArgs =
     finalAttrs:
     {
       enableUpdateScript ? true,
@@ -145,5 +149,4 @@ let
         ''
         + prevAttrs.postInstall or "";
     };
-in
-stdenv.mkDerivation (lib.extends postgresqlBuildExtension (lib.toFunction args))
+}