diff --git a/tests/python/frontend/tensorflow/test_forward.py b/tests/python/frontend/tensorflow/test_forward.py
index e952698880b0c..6b9558ea47c77 100644
--- a/tests/python/frontend/tensorflow/test_forward.py
+++ b/tests/python/frontend/tensorflow/test_forward.py
@@ -3628,7 +3628,7 @@ def _test_forward_segment(name, data_shape, size):
         tf_segment_ids = tf.constant(segment_ids)
         segment_op = getattr(tf.math, name)
         result = segment_op(in_data, tf_segment_ids, name=name)
-        compare_tf_with_tvm([np_data, segment_ids], ["in_data:0"], result.name, False)
+        compare_tf_with_tvm([np_data, segment_ids], ["in_data:0"], result.name, True)
 
     _test_forward_segment("segment_max", [100], 20)
     _test_forward_segment("segment_max", [10, 3, 4], 5)
@@ -3674,7 +3674,7 @@ def _test_forward_unsorted_segment(name, data_shape, size):
         tf_segment_ids = tf.constant(segment_ids)
         segment_op = getattr(tf.math, name)
         result = segment_op(in_data, tf_segment_ids, num_segments, name=name)
-        compare_tf_with_tvm([np_data, segment_ids], ["in_data:0"], result.name, False)
+        compare_tf_with_tvm([np_data, segment_ids], ["in_data:0"], result.name, True)
 
     _test_forward_unsorted_segment("unsorted_segment_max", [100], 20)
     _test_forward_unsorted_segment("unsorted_segment_max", [10, 3, 4], 5)