diff --git a/tests/hermes_cli/test_update_yes_flag.py b/tests/hermes_cli/test_update_yes_flag.py
index 1e10ddb5d3496..20d99e8b08165 100644
--- a/tests/hermes_cli/test_update_yes_flag.py
+++ b/tests/hermes_cli/test_update_yes_flag.py
@@ -181,9 +181,7 @@ class TestUnicodeDecodeErrorInUpdatePrompts:
 
         out = capsys.readouterr().out
         assert "hermes config migrate" in out
-        assert mock_migrate.call_count == 0 or (
-            mock_migrate.call_args and mock_migrate.call_args.kwargs.get("interactive") is not True
-        )
+        mock_migrate.assert_not_called()
 
     def test_stash_restore_unicode_decode_error_falls_through_to_skip(self, tmp_path, capsys):
         from hermes_cli.update_cmd import _restore_stashed_changes