diff --git a/parts/lib/functions.nix b/parts/lib/functions.nix
index 3555c72..75aa683 100644
--- a/parts/lib/functions.nix
+++ b/parts/lib/functions.nix
@@ -8,7 +8,6 @@ let
   inherit (lib) mkOption;
 in
 {
-
   # Verify the existence of a binary inside of a derivation.
   # Returns the path to the binary or throws.
   checkPath =
@@ -92,7 +91,7 @@ in
       };
     };
     mkSwap = size: {
-      size = "32G";
+      inherit size;
       content = {
         type = "swap";
         randomEncryption = true;