1 files changed, 9 insertions, 1 deletions

diff --git a/tests/test_git_trace2_event_log.py b/tests/test_git_trace2_event_log.py
index 7bd21e21..89dcfb92 100644
--- a/tests/test_git_trace2_event_log.py
+++ b/tests/test_git_trace2_event_log.py
@@ -66,6 +66,13 @@ class EventLogTestCase(unittest.TestCase):
         log_data.append(json.loads(line))
     return log_data
 
+  def remove_prefix(self, s, prefix):
+    """Return a copy string after removing |prefix| from |s|, if present or the original string."""
+    if s.startswith(prefix):
+        return s[len(prefix):]
+    else:
+        return s
+
   def test_initial_state_with_parent_sid(self):
     """Test initial state when 'GIT_TRACE2_PARENT_SID' is set by parent."""
     self.assertRegex(self._event_log_module.full_sid, self.FULL_SID_REGEX)
@@ -265,7 +272,8 @@ class EventLogTestCase(unittest.TestCase):
       # Check for 'data' event specific fields.
       self.assertIn('key', event)
       self.assertIn('value', event)
-      key = event['key'].removeprefix(f'{prefix_value}/')
+      key = event['key']
+      key = self.remove_prefix(key, f'{prefix_value}/')
       value = event['value']
       self.assertEqual(self._event_log_module.GetDataEventName(value), event['event'])
       self.assertTrue(key in config and value == config[key])