diff --git a/sdks/python/apache_beam/runners/interactive/recording_manager_test.py b/sdks/python/apache_beam/runners/interactive/recording_manager_test.py
index 55c3bd91cfd6..b43d8aeb8731 100644
--- a/sdks/python/apache_beam/runners/interactive/recording_manager_test.py
+++ b/sdks/python/apache_beam/runners/interactive/recording_manager_test.py
@@ -1139,7 +1139,10 @@ def test_get_pipeline_graph_not_cached(self):
         t.unique_name
         for t in graph2._pipeline_proto.components.transforms.values()
     ]
-    self.assertIn('Map1', transform_names)
+    self.assertTrue(
+        any('Map1' in name for name in transform_names),
+        f"Expected 'Map1' in one of the transform names, got: {transform_names}"
+    )
 
   def test_wait_for_completion_raises_exception_on_failure(self):
     future = Future()