diff --git a/python/paddle/fluid/tests/unittests/test_data_balance.py b/python/paddle/fluid/tests/unittests/test_data_balance.py
index aa19a5edc7814315edaacf6e76072f62fcf7eb55..9a6b7cf4765eb28d85d6f9accb1682b57df2ff07 100644
--- a/python/paddle/fluid/tests/unittests/test_data_balance.py
+++ b/python/paddle/fluid/tests/unittests/test_data_balance.py
@@ -194,4 +194,6 @@ class TestDataBalance(unittest.TestCase):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    # Disable data balance unittest, because data balance would be removed
+    # unittest.main()
+    pass