diff --git a/tensorflow/python/distribute/collective_all_reduce_strategy_test.py b/tensorflow/python/distribute/collective_all_reduce_strategy_test.py
index 14ed6b071d1..0c0bff429e6 100644
--- a/tensorflow/python/distribute/collective_all_reduce_strategy_test.py
+++ b/tensorflow/python/distribute/collective_all_reduce_strategy_test.py
@@ -364,9 +364,9 @@ class CollectiveAllReduceStrategyTestBase(
         computed_value = sess.run([values.select_replica(r, next_element)
                                    for r in range(len(devices))])
         if ignore_order:
-          self.assertCountEqual(expected_value, computed_value)
+          self.assertCountEqual(list(expected_value), list(computed_value))
         else:
-          self.assertEqual(expected_value, computed_value)
+          self.assertEqual(list(expected_value), list(computed_value))
 
       with self.assertRaises(errors.OutOfRangeError):
         next_element = iterator.get_next()
@@ -382,9 +382,9 @@ class CollectiveAllReduceStrategyTestBase(
           computed_value = sess.run([values.select_replica(r, next_element)
                                      for r in range(len(devices))])
           if ignore_order:
-            self.assertCountEqual(expected_value, computed_value)
+            self.assertCountEqual(list(expected_value), list(computed_value))
           else:
-            self.assertEqual(expected_value, computed_value)
+            self.assertEqual(list(expected_value), list(computed_value))
 
 
 class DistributedCollectiveAllReduceStrategyTest(