diff --git a/armada/handlers/wait.py b/armada/handlers/wait.py
index b79d74d3..d74353ed 100644
--- a/armada/handlers/wait.py
+++ b/armada/handlers/wait.py
@@ -725,6 +725,7 @@ class StatefulSetWait(ControllerWait):
                         msg.format(ASYNC_UPDATE_NOT_ALLOWED_MSG, strategy))
 
                 if (is_rolling and replicas and
+                        spec.update_strategy.rolling_update and
                         spec.update_strategy.rolling_update.partition):
                     msg = "{}: partitioned rollout"