diff --git a/python/paddle/fluid/tests/unittests/test_data_balance.py b/python/paddle/fluid/tests/unittests/test_data_balance.py
index 44c1adad9e3138f3ed327494e16875e0104b0f3f..b558d7c2ea172d9c7526c865a4bc54c32f8998b6 100644
--- a/python/paddle/fluid/tests/unittests/test_data_balance.py
+++ b/python/paddle/fluid/tests/unittests/test_data_balance.py
@@ -15,7 +15,6 @@
 import unittest
 import paddle.fluid as fluid
 import paddle.v2 as paddle
-import paddle.v2.dataset.mnist as mnist
 import numpy as np