diff --git a/python/paddle/v2/__init__.py b/python/paddle/v2/__init__.py
index 25526bf409cf82f26979a84700ce948ac969df0c..2c5daeb9b096bdccb8d9e080bc9b8731fbe08aaf 100644
--- a/python/paddle/v2/__init__.py
+++ b/python/paddle/v2/__init__.py
@@ -23,17 +23,19 @@ import data_feeder
 import networks
 from . import dataset
 from . import reader
+from . import plot
 import attr
 import pooling
 import inference
 import networks
 import py_paddle.swig_paddle as api
 import minibatch
+import plot
 
 __all__ = [
     'optimizer', 'layer', 'activation', 'parameters', 'init', 'trainer',
     'event', 'data_type', 'attr', 'pooling', 'data_feeder', 'dataset', 'reader',
-    'topology', 'networks', 'infer'
+    'topology', 'networks', 'infer', 'plot'
 ]
 
 
diff --git a/python/setup.py.in b/python/setup.py.in
index 68ca35265cf13265ad0b171b0f70e20b83006ff9..4ac35e3b8d6049a9024a6e0c9bb6804900f82197 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -7,7 +7,8 @@ packages=['paddle',
           'paddle.utils',
           'paddle.v2',
           'paddle.v2.dataset',
-          'paddle.v2.reader']
+          'paddle.v2.reader',
+          'paddle.v2.plot']
 
 setup(name='paddle',
       version='${PADDLE_VERSION}',