diff --git a/lib/modules/partition/disks.nix b/lib/modules/partition/disks.nix
index a0dd0947..dd125fdb 100644
--- a/lib/modules/partition/disks.nix
+++ b/lib/modules/partition/disks.nix
@@ -189,7 +189,7 @@ let
         script = lib.mkDefault ''
           function partition() {
             ${if (!config.wipe) then cleanup else ""}
-
+            ${probeScript}
             ${fdiskCommands}
             ${fixType}
           }