diff --git a/tools/image_builder/__main__.py b/tools/image_builder/__main__.py
index 6886dc8be..61ca37418 100644
--- a/tools/image_builder/__main__.py
+++ b/tools/image_builder/__main__.py
@@ -114,11 +114,9 @@ def build_image(
         load=True,
         platforms=[target_platform],
         tags=docker_tags,
+        push=push,
     )
 
-    if push:
-        docker.push(docker_tags)
-
 
 @click.command()
 @click.option(