diff --git a/docs/development/adding_a_new_op.md b/docs/development/adding_a_new_op.md
index bb74f12c6a668addf49d409288898e4f27377c20..eda612d34739731b3431d8d2657ef7d693c619c3 100644
--- a/docs/development/adding_a_new_op.md
+++ b/docs/development/adding_a_new_op.md
@@ -58,7 +58,7 @@ Define the Ops registering function in `mace/ops/my_custom_op.cc`.
 namespace mace {
 namespace ops {
 
-void Register_My_Custom_Op(OperatorRegistry *op_registry) {
+void Register_My_Custom_Op(OperatorRegistryBase *op_registry) {
   REGISTER_OPERATOR(op_registry, OpKeyBuilder("my_custom_op")
                                      .Device(DeviceType::CPU)
                                      .TypeConstraint<float>("T")