diff --git a/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py b/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py
index 5a64d2667f2a..904b90710225 100644
--- a/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py
+++ b/sdks/python/apache_beam/examples/snippets/transforms/elementwise/enrichment_test.py
@@ -130,8 +130,8 @@ def test_enrichment_with_bigtable(self, mock_stdout):
 
   def test_enrichment_with_vertex_ai(self, mock_stdout):
     enrichment_with_vertex_ai()
-    output = mock_stdout.getvalue().splitlines()
-    expected = validate_enrichment_with_vertex_ai()
+    output = sorted(mock_stdout.getvalue().splitlines())
+    expected = sorted(validate_enrichment_with_vertex_ai())
 
     for i in range(len(expected)):
       self.assertEqual(set(output[i].split(',')), set(expected[i].split(',')))