diff --git a/tests/test_update_banner_fixes.py b/tests/test_update_banner_fixes.py
index da6b4000..73df0cc4 100644
--- a/tests/test_update_banner_fixes.py
+++ b/tests/test_update_banner_fixes.py
@@ -863,7 +863,7 @@ class TestAgentUpdateRequiresGatewayRestart:
                 return '', False
             if args[0] == 'stash':
                 return '', True
-            if args[:2] == ['reset', '--hard', 'HEAD']:
+            if args[:3] == ['reset', '--hard', 'HEAD']:
                 return '', True
             if args[0] == 'pull':
                 return 'Updating', True