diff --git a/pkgs/stdenv/generic/make-derivation.nix b/pkgs/stdenv/generic/make-derivation.nix
index 5fae0012db0e..e9760ee1be27 100644
--- a/pkgs/stdenv/generic/make-derivation.nix
+++ b/pkgs/stdenv/generic/make-derivation.nix
@@ -24,7 +24,6 @@ let
     isAttrs
     isBool
     isDerivation
-    isFunction
     isInt
     isList
     isString
@@ -37,6 +36,7 @@ let
     remove
     splitString
     subtractLists
+    toFunction
     unique
     zipAttrsWith
     ;
@@ -62,14 +62,7 @@ let
     Most arguments are also passed through to the underlying call of [`builtins.derivation`](https://nixos.org/manual/nix/stable/language/derivations).
     :::
   */
-  mkDerivation =
-    fnOrAttrs:
-    makeDerivationExtensible (
-      if isFunction fnOrAttrs then
-        fnOrAttrs
-      else
-        (_: fnOrAttrs)
-    );
+  mkDerivation = fnOrAttrs: makeDerivationExtensible (toFunction fnOrAttrs);
 
   checkMeta = import ./check-meta.nix {
     inherit lib config;