diff --git a/python/paddle/fluid/tests/unittests/test_teacher_student_sigmoid_loss_op.py b/python/paddle/fluid/tests/unittests/test_teacher_student_sigmoid_loss_op.py
index a58ef8969c0ffff8cc5deb902bfbdf5438064d8f..f133d470c54413aa90992d23d6917ca1f60b56ba 100644
--- a/python/paddle/fluid/tests/unittests/test_teacher_student_sigmoid_loss_op.py
+++ b/python/paddle/fluid/tests/unittests/test_teacher_student_sigmoid_loss_op.py
@@ -28,7 +28,7 @@ class TestTeacherStudentSigmoidLossOp(OpTest):
 
     def setUp(self):
         self.op_type = "teacher_student_sigmoid_loss"
-        batch_size = 16
+        batch_size = 100
         num_classes = 1
         self.inputs = {
             'X': logit(
diff --git a/python/paddle/fluid/tests/unittests/test_tree_conv_op.py b/python/paddle/fluid/tests/unittests/test_tree_conv_op.py
index 8feb2333f7b52dc4a428c59b21c3ea543bfaaef9..8261e31bceb7a5df4c97d6ef8bebc724d2dbbd98 100644
--- a/python/paddle/fluid/tests/unittests/test_tree_conv_op.py
+++ b/python/paddle/fluid/tests/unittests/test_tree_conv_op.py
@@ -54,7 +54,7 @@ class TestTreeConvOp(OpTest):
         self.fea_size = 3
         self.output_size = 1
         self.max_depth = 2
-        self.batch_size = 1
+        self.batch_size = 2
         self.num_filters = 1
         adj_array = [
             1, 2, 1, 3, 1, 4, 1, 5, 2, 6, 2, 7, 2, 8, 4, 9, 4, 10, 5, 11, 6, 12,
diff --git a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py
index 62f2668a773a7812c7d5c5c912ebcb0dcb5ebbe4..11d08970b5636a05b40df941f39112ee96bdb68f 100644
--- a/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py
+++ b/python/paddle/fluid/tests/unittests/white_list/check_shape_white_list.py
@@ -20,6 +20,5 @@ NOT_CHECK_OP_LIST = [
 NEED_TO_FIX_OP_LIST = [
     'sequence_scatter', 'log_loss', 'sequence_topk_avg_pooling', 'matmul',
     'add_position_encoding', 'margin_rank_loss', 'elementwise_pow',
-    'fused_elemwise_activation', 'tree_conv', 'mul',
-    'teacher_student_sigmoid_loss'
+    'fused_elemwise_activation', 'mul'
 ]