diff --git a/internal/agent/runtime/docker_test.go b/internal/agent/runtime/docker_test.go
index 4237d6cbe..3a38ae7e9 100644
--- a/internal/agent/runtime/docker_test.go
+++ b/internal/agent/runtime/docker_test.go
@@ -27,7 +27,7 @@ func TestDockerImageNotPresent(t *testing.T) {
 		t.Fatalf("Received unexpected error: %v", err)
 	}
 
-	image := "alpine"
+	image := "hello-world"
 
 	images, err := clnt.ImageList(context.Background(), types.ImageListOptions{
 		Filters: filters.NewArgs(filters.Arg("reference", image)),
diff --git a/internal/workflow/template_funcs.go b/internal/workflow/template_funcs.go
index c6af8e5d8..46d39fa5c 100644
--- a/internal/workflow/template_funcs.go
+++ b/internal/workflow/template_funcs.go
@@ -20,11 +20,12 @@ var templateFuncs = map[string]interface{}{
 //
 //	formatPartition("/dev/nvme0n1", 0) -> /dev/nvme0n1p1
 //	formatPartition("/dev/sda", 1) -> /dev/sda1
+//	formatPartition("/dev/vda", 2) -> /dev/vda2
 func formatPartition(dev string, partition int) string {
 	switch {
 	case strings.HasPrefix(dev, "/dev/nvme"):
 		return fmt.Sprintf("%vp%v", dev, partition)
-	case strings.HasPrefix(dev, "/dev/sd"):
+	case strings.HasPrefix(dev, "/dev/sd"), strings.HasPrefix(dev, "/dev/vd"), strings.HasPrefix(dev, "/dev/hd"):
 		return fmt.Sprintf("%v%v", dev, partition)
 	}
 	return dev