diff --git a/docker/kolla-ansible/Dockerfile.j2 b/docker/kolla-ansible/Dockerfile.j2
index 0e6770ff4c743c98ba35e1377d14564d84677122..1d0228d548725fdadad2fa485ffebe3a6fcfbf80 100644
--- a/docker/kolla-ansible/Dockerfile.j2
+++ b/docker/kolla-ansible/Dockerfile.j2
@@ -49,7 +49,7 @@ RUN mkdir -p /etc/ansible /usr/share/ansible /home/ansible \
     && echo 'localhost ansible_connection=local' > /etc/ansible/hosts \
     && useradd --user-group ansible --groups kolla
 
-COPY find_disks.py kolla_keystone_service.py kolla_keystone_user.py kolla_sanity.py /usr/share/ansible/
+COPY find_disks.py kolla_keystone_service.py kolla_keystone_user.py kolla_sanity.py kolla_zookeeper.py /usr/share/ansible/
 COPY ansible.cfg /home/ansible/.ansible.cfg
 
 {{ include_footer }}
diff --git a/docker/kolla-ansible/kolla_zookeeper.py b/docker/kolla-ansible/kolla_zookeeper.py
index da692605a56b3a3f8ddf706a026bf53677930ece..5362000762d978a68305cf3866e6efc1c35480a1 100644
--- a/docker/kolla-ansible/kolla_zookeeper.py
+++ b/docker/kolla-ansible/kolla_zookeeper.py
@@ -16,7 +16,8 @@
 
 import contextlib
 
-import kazoo
+import kazoo.client
+import kazoo.exceptions
 
 
 @contextlib.contextmanager
@@ -49,9 +50,9 @@ def main():
                 zk.get(path)
             except kazoo.exceptions.NoNodeError:
                 if value is None:
-                    zk.create(path)
+                    zk.create(path, makepath=True)
                 else:
-                    zk.create(path, value=value.encode())
+                    zk.create(path, value=value.encode(), makepath=True)
                 changed = True
 
         module.exit_json(changed=changed)