diff --git a/nixos/lib/make-disk-image.nix b/nixos/lib/make-disk-image.nix
index 4806fe9dbac71..8aa606a56af80 100644
--- a/nixos/lib/make-disk-image.nix
+++ b/nixos/lib/make-disk-image.nix
@@ -71,7 +71,7 @@ let format' = format; in let
     vdi   = "vdi";
     vpc   = "vhd";
     raw   = "img";
-  }.${format};
+  }.${format} or format;
 
   rootPartition = { # switch-case
     legacy = "1";