diff --git a/tests/upgrade.sh b/tests/upgrade.sh
index 898787e707f340fc056395442b02816b9aa15138..c0ce10e441e98fbad4f4c6c7e0320a5519862452 100755
--- a/tests/upgrade.sh
+++ b/tests/upgrade.sh
@@ -10,10 +10,8 @@ export PYTHONUNBUFFERED=1
 function upgrade {
     RAW_INVENTORY=/etc/kolla/inventory
 
-    # TODO(jeffrey4l): need run a real upgrade
     tools/kolla-ansible -i ${RAW_INVENTORY} -vvv prechecks &> /tmp/logs/ansible/upgrade-prechecks
-    # TODO(mgoddard): add pull action when we have a local registry service in
-    # CI.
+    tools/kolla-ansible -i ${RAW_INVENTORY} -vvv pull &> /tmp/logs/ansible/pull-upgrade
     tools/kolla-ansible -i ${RAW_INVENTORY} -vvv upgrade &> /tmp/logs/ansible/upgrade
     tools/kolla-ansible -i ${RAW_INVENTORY} -vvv check &> /tmp/logs/ansible/check-upgrade
 }