diff --git a/python/paddle/fluid/tests/unittests/test_sequence_pad_op.py b/python/paddle/fluid/tests/unittests/test_sequence_pad_op.py
index 3ac7371aa70e60d7d97d8a5274d124f4e1e72d5a..3067294e5bb3edcb2f1ce77f5e60b885a39a6475 100644
--- a/python/paddle/fluid/tests/unittests/test_sequence_pad_op.py
+++ b/python/paddle/fluid/tests/unittests/test_sequence_pad_op.py
@@ -62,7 +62,7 @@ class TestSequencePadOp(OpTest):
             start_idx = end_idx
 
         out_data = np.array(padded_sequences)
-        length = np.array(self.x_len_lod[0])
+        length = np.array(self.x_len_lod[0]).reshape((-1, 1))
         self.outputs = {'Out': out_data, 'Length': length}
 
     def setUp(self):
@@ -130,3 +130,7 @@ class TestSequencePadOp7(TestSequencePadOp):
         self.pad_value = [1.0]
         self.padded_length = 7
         self.dtype = 'float32'
+
+
+if __name__ == '__main__':
+    unittest.main()