diff --git a/swebench/harness/docker_build.py b/swebench/harness/docker_build.py
index 3f71ed84..d7bcf8ee 100644
--- a/swebench/harness/docker_build.py
+++ b/swebench/harness/docker_build.py
@@ -241,7 +241,7 @@ def get_env_configs_to_build(
                 # Remove the environment image if it was built after the base_image
                 for dep in find_dependent_images(client, test_spec.env_image_key):
                     # Remove instance images that depend on this environment image
-                    remove_image(client, dep.image_id, "quiet")
+                    remove_image(client, dep, "quiet")
                 remove_image(client, test_spec.env_image_key, "quiet")
                 image_exists = False
         except docker.errors.ImageNotFound: