diff --git a/stub-images/vddk-test-vmdk/BUILD.bazel b/stub-images/vddk-test-vmdk/BUILD.bazel
index 630a300..5a1f689 100644
--- a/stub-images/vddk-test-vmdk/BUILD.bazel
+++ b/stub-images/vddk-test-vmdk/BUILD.bazel
@@ -7,7 +7,7 @@ container_image(
     files = [
         "@cirros//file",
     ],
-    base = "//virt-v2v:forklift-virt-v2v",
+    base = "//virt-v2v/cold:forklift-virt-v2v",
     user = "1001",
     visibility = ["//visibility:public"],
 )
diff --git a/stub-images/virt-v2v-stub/BUILD.bazel b/stub-images/virt-v2v-stub/BUILD.bazel
index 3c8e1d9..76c3acd 100644
--- a/stub-images/virt-v2v-stub/BUILD.bazel
+++ b/stub-images/virt-v2v-stub/BUILD.bazel
@@ -5,7 +5,7 @@ load(
 
 container_image(
     name = "forklift-virt-v2v-stub",
-    base = "//virt-v2v:forklift-virt-v2v",
+    base = "//virt-v2v/cold:forklift-virt-v2v",
     directory = "/usr/local/bin/",
     empty_dirs = ["/disks"],
     entrypoint = ["/usr/local/bin/entrypoint"],