diff --git a/puppet/services/docker-registry.yaml b/puppet/services/docker-registry.yaml
index faeeda59c3..b62e433015 100644
--- a/puppet/services/docker-registry.yaml
+++ b/puppet/services/docker-registry.yaml
@@ -67,11 +67,10 @@ outputs:
       upgrade_tasks:
         - block:
             - name: Install docker packages on upgrade if missing
-              when: step|int == 3
               package: name=docker-distribution state=latest
             - set_fact: *docker_vars
             - name: Reconfigure Docker if needed
               include_role:
                 name: container-registry
                 tasks_from: docker
-          when: step|int = 3
+          when: step|int == 3