diff --git a/paddle/framework/CMakeLists.txt b/paddle/framework/CMakeLists.txt
index d9c84f3c0a7fe6fd307614583a5ebc07d0774512..1ba23a2c3f46a67b6261f838a377d321f3e433ec 100644
--- a/paddle/framework/CMakeLists.txt
+++ b/paddle/framework/CMakeLists.txt
@@ -50,7 +50,7 @@ cc_library(executor SRCS executor.cc DEPS op_registry device_context scope frame
 #endif()
 
 cc_library(prune SRCS prune.cc)
-cc_test(prune_test SRCS prune_test.cc DEPS prune recurrent_op device_context)
+cc_test(prune_test SRCS prune_test.cc DEPS op_info prune recurrent_op device_context)
 
 cc_library(tensor_array SRCS tensor_array.cc DEPS lod_tensor)
 cc_test(tensor_array_test SRCS tensor_array_test.cc DEPS tensor_array place)
diff --git a/paddle/framework/prune_test.cc b/paddle/framework/prune_test.cc
index 790fa169244198c41b42d81623504f268f2c91c8..c351c12d2213b061ed624285cb1ae8098eaf8f7a 100644
--- a/paddle/framework/prune_test.cc
+++ b/paddle/framework/prune_test.cc
@@ -54,8 +54,6 @@ class TwoOneOpMaker : public OpProtoAndCheckerMaker {
 
 namespace f = paddle::framework;
 namespace ops = paddle::operators;
-REGISTER_OP_WITHOUT_GRADIENT(one_one, f::NOP, f::OneOneOpMaker);
-REGISTER_OP_WITHOUT_GRADIENT(two_one, f::NOP, f::TwoOneOpMaker);
 
 void AddOp(const std::string &type, const f::VariableNameMap &inputs,
            const f::VariableNameMap &outputs, f::AttributeMap attrs,