diff --git a/python/paddle/v2/plot/plot.py b/python/paddle/v2/plot/plot.py
index d36fc47d8e2be0b90ea0f0db41430b94edcd7b8b..a685036970284684112e0c2b75d2705c26f15003 100644
--- a/python/paddle/v2/plot/plot.py
+++ b/python/paddle/v2/plot/plot.py
@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from IPython import display
 import os
 
 
@@ -39,7 +38,9 @@ class Ploter(object):
         self.__disable_plot__ = os.environ.get("DISABLE_PLOT")
         if not self.__plot_is_disabled__():
             import matplotlib.pyplot as plt
+            from IPython import display
             self.plt = plt
+            self.display = display
 
     def __plot_is_disabled__(self):
         return self.__disable_plot__ == "True"
@@ -63,8 +64,8 @@ class Ploter(object):
                 titles.append(title)
                 self.plt.plot(data.step, data.value)
         self.plt.legend(titles, loc='upper left')
-        display.clear_output(wait=True)
-        display.display(self.plt.gcf())
+        self.display.clear_output(wait=True)
+        self.display.display(self.plt.gcf())
         self.plt.gcf().clear()
 
     def reset(self):
diff --git a/python/paddle/v2/plot/tests/run_tests.sh b/python/paddle/v2/plot/tests/run_tests.sh
index 20514160cd0e46c42d7940629274c6254ccc9884..9c1a4a71ce43f285c4f970eddf6af46a2821a40a 100755
--- a/python/paddle/v2/plot/tests/run_tests.sh
+++ b/python/paddle/v2/plot/tests/run_tests.sh
@@ -21,6 +21,7 @@ popd > /dev/null
 cd $SCRIPTPATH
 $1 -m pip install ../../../../../paddle/dist/*.whl
 
+export DISABLE_PLOT="True"
 test_list="test_ploter.py"
 
 export PYTHONPATH=$PWD/../../../../../python/