Partition drives after action setup

scripts/Common/OS/setup.fish

@@ -39,6 +39,15 @@ function runSetup
         end
 
         initBackupConfig --action restore
+        echo "Partitioning drives..."
+
+        begin
+            set -l script (mktemp)
+            chmod +x "$script"
+            and getOSConfig partition.script >"$script"
+            and "$script"
+            and rm "$script"
+        end
 
         if [ -n "$VALHALLA_BACKUP_SERVER_KEY" ]
             set -l knownHosts /root/.ssh/known_hosts
@@ -77,15 +86,6 @@ function runSetup
         end | runChroot "$mountDir" tee /root/.bash_profile >/dev/null
     end
 
-    function actionPreRun
-        set -l script (mktemp)
-        echo "Partitioning drives..."
-        chmod +x "$script"
-        and getOSConfig partition.script >"$script"
-        and "$script"
-        and rm "$script"
-    end
-
     function actionPostRun
         echo "Setup finished!"
         and echo "This machine will reboot in 5 seconds..."