diff --git a/python/paddle/v2/data_converter.py b/python/paddle/v2/data_converter.py
index 5d7b8a736b3596e0d19a4607ad1f22b849b0cc21..45114b407dae2207d3b5df3bf2ea8a5883c7f332 100644
--- a/python/paddle/v2/data_converter.py
+++ b/python/paddle/v2/data_converter.py
@@ -15,6 +15,7 @@
 import collections
 import py_paddle.swig_paddle
 import numpy
+import paddle.trainer.PyDataProvider2 as dp2
 
 __all__ = ['DataConverter']