diff --git a/sdks/python/apache_beam/transforms/util_test.py b/sdks/python/apache_beam/transforms/util_test.py
index 18efc44e4d34..d834a1c5efea 100644
--- a/sdks/python/apache_beam/transforms/util_test.py
+++ b/sdks/python/apache_beam/transforms/util_test.py
@@ -231,11 +231,11 @@ def test_reshuffle_timestamps_unchanged(self):
   def test_reshuffle_windows_unchanged(self):
     pipeline = TestPipeline()
     data = [(1, 1), (2, 1), (3, 1), (1, 2), (2, 2), (1, 4)]
-    expected_data = [TestWindowedValue(v, t, [w]) for (v, t, w) in
-                     [((1, [2, 1]), 4.0, IntervalWindow(1.0, 4.0)),
-                      ((2, [2, 1]), 4.0, IntervalWindow(1.0, 4.0)),
-                      ((3, [1]), 3.0, IntervalWindow(1.0, 3.0)),
-                      ((1, [4]), 6.0, IntervalWindow(4.0, 6.0))]]
+    expected_data = [TestWindowedValue(v, t, [w]) for (v, t, w) in [
+        ((1, contains_in_any_order([2, 1])), 4.0, IntervalWindow(1.0, 4.0)),
+        ((2, contains_in_any_order([2, 1])), 4.0, IntervalWindow(1.0, 4.0)),
+        ((3, [1]), 3.0, IntervalWindow(1.0, 3.0)),
+        ((1, [4]), 6.0, IntervalWindow(4.0, 6.0))]]
     before_reshuffle = (pipeline
                         | 'start' >> beam.Create(data)
                         | 'add_timestamp' >> beam.Map(