diff --git a/image-yaml/overcloud-images.yaml b/image-yaml/overcloud-images.yaml
index 124d1c5c0..ab2b9cf15 100644
--- a/image-yaml/overcloud-images.yaml
+++ b/image-yaml/overcloud-images.yaml
@@ -37,6 +37,8 @@ disk_images:
       - python-heat-agent-json-file
     options:
       - "--min-tmpfs 5"
+    environment:
+      DIB_PYTHON_VERSION: '2'
   -
     imagename: ironic-python-agent
     arch: amd64
@@ -56,4 +58,5 @@ disk_images:
       - python-hardware-detect
     options:
       - "--min-tmpfs=5"
-
+    environment:
+      DIB_PYTHON_VERSION: '2'