diff --git a/python/paddle/v2/layer.py b/python/paddle/v2/layer.py
index 6dea3afbcf407fd5490f3d9c091dda07f7db5f42..dbd802bee8270dc96056bb299dc835c54b128c6f 100644
--- a/python/paddle/v2/layer.py
+++ b/python/paddle/v2/layer.py
@@ -139,6 +139,8 @@ class Layer(object):
 
         if self.name is None:
             return self.to_proto_impl(**kwargs)
+        elif isinstance(self, MemoryV2):
+            return self.to_proto_impl(**kwargs)
         elif self.name not in context:
             context[self.name] = self.to_proto_impl(**kwargs)
 
diff --git a/python/paddle/v2/tests/CMakeLists.txt b/python/paddle/v2/tests/CMakeLists.txt
index bc0f24792796b82d0c4c59addb2ee5457b452ae2..b2f43c42de8ebfe82dd16fa6b0667b4c7bd59370 100644
--- a/python/paddle/v2/tests/CMakeLists.txt
+++ b/python/paddle/v2/tests/CMakeLists.txt
@@ -1,7 +1,11 @@
 add_test(NAME test_v2_layer
         COMMAND ${PROJ_ROOT}/paddle/.set_python_path.sh -d ${PROJ_ROOT}/python/
-        ${PYTHON_EXECUTABLE} ${PROJ_ROOT}/python/paddle/v2/tests/test_layer.py)
+        ${PYTHON_EXECUTABLE} ${PROJ_ROOT}/python/paddle/v2/tests/test_layer.py
+        WORKING_DIRECTORY ${PROJ_ROOT}/python/paddle)
 
 add_test(NAME test_v2_rnn_layer
         COMMAND ${PROJ_ROOT}/paddle/.set_python_path.sh -d ${PROJ_ROOT}/python/
         ${PYTHON_EXECUTABLE} ${PROJ_ROOT}/python/paddle/v2/tests/test_rnn_layer.py)
+
+add_test(NAME test_v2_api
+        COMMAND bash ${PROJ_ROOT}/python/paddle/v2/tests/run_tests.sh ${PYTHON_EXECUTABLE})