diff --git a/toolkit/tools/pkg/imagecustomizerlib/imagecustomizer.go b/toolkit/tools/pkg/imagecustomizerlib/imagecustomizer.go
index bdc7993fd5e..fcdf089d913 100644
--- a/toolkit/tools/pkg/imagecustomizerlib/imagecustomizer.go
+++ b/toolkit/tools/pkg/imagecustomizerlib/imagecustomizer.go
@@ -390,9 +390,11 @@ func customizeOSContents(ic *ImageCustomizerParameters) error {
 		}
 	}
 
-	err = customizeUkiImageHelper(ic.buildDirAbs, ic.rawImageFile)
-	if err != nil {
-		return err
+	if ic.config.OS.Uki {
+		err = customizeUkiImageHelper(ic.buildDirAbs, ic.rawImageFile)
+		if err != nil {
+			return err
+		}
 	}
 
 	// Check file systems for corruption.