diff --git a/vintage.py b/vintage.py
index 6f390e5..5c03361 100644
--- a/vintage.py
+++ b/vintage.py
@@ -169,7 +169,7 @@ def on_query_context(self, view, key, operator, operand, match_all):
             if operator == sublime.OP_NOT_EQUAL:
                 return m != g_input_state.motion_mode
         elif key == "vi_has_repeat_digit":
-            if g_input_state.motion_command:
+            if g_input_state.action_command:
                 v = len(g_input_state.motion_repeat_digits) > 0
             else:
                 v = len(g_input_state.prefix_repeat_digits) > 0