diff --git a/lib/modules/partition.nix b/lib/modules/partition.nix
index a6d670d9..ecbb92af 100644
--- a/lib/modules/partition.nix
+++ b/lib/modules/partition.nix
@@ -264,28 +264,6 @@
 
     config = {
       partition = {
-        os = {
-          partitions = {
-            Boot = {
-              index = 1;
-              type = "uefi";
-              size = "+1G";
-              format = fs.fat32;
-            };
-
-            Swap = {
-              index = 2;
-              type = "swap";
-            };
-
-            OS = {
-              index = 3;
-              type = "linux";
-              format = fs.ext4;
-            };
-          };
-        };
-
         script =
           let
             inherit (config.partition) os rootDir;
@@ -314,6 +292,28 @@
               swapScript
             ]
           );
+
+        os = {
+          partitions = {
+            Boot = {
+              index = 1;
+              type = "uefi";
+              size = "+1G";
+              format = fs.fat32;
+            };
+
+            Swap = {
+              index = 2;
+              type = "swap";
+            };
+
+            OS = {
+              index = 3;
+              type = "linux";
+              format = fs.ext4;
+            };
+          };
+        };
       };
     };
   }