diff --git a/cmd/main.go b/cmd/main.go
index 3e6e52f..7c1c360 100644
--- a/cmd/main.go
+++ b/cmd/main.go
@@ -100,7 +100,7 @@ func setupEnv() int {
 		Drives: []models.Drive{
 			{
 				DriveID:      firecracker.String("rootfs"),
-				PathOnHost:   firecracker.String("./out/ubuntu-22.04.ext4"),
+				PathOnHost:   firecracker.String("./out/rootfs.ext4"),
 				IsReadOnly:   firecracker.Bool(false),
 				IsRootDevice: firecracker.Bool(true),
 			},