diff --git a/plugin/controller/lib/AppLoadUnitControllerHook.ts b/plugin/controller/lib/AppLoadUnitControllerHook.ts
index 5db156b5..ea25f251 100644
--- a/plugin/controller/lib/AppLoadUnitControllerHook.ts
+++ b/plugin/controller/lib/AppLoadUnitControllerHook.ts
@@ -25,7 +25,7 @@ export class AppLoadUnitControllerHook implements LifecycleHook<LoadUnitLifecycl
         throw new Error(`not find controller implement for ${String(proto.name)} which type is ${metadata.type}`);
       }
       ControllerMetadataManager.instance.addController(metadata);
-      await register.register(this.rootProtoManager);
+      await register.register(this.rootProtoManager, obj);
     }
   }
 }
diff --git a/plugin/controller/lib/ControllerRegister.ts b/plugin/controller/lib/ControllerRegister.ts
index 8bc3bde7..a402533a 100644
--- a/plugin/controller/lib/ControllerRegister.ts
+++ b/plugin/controller/lib/ControllerRegister.ts
@@ -1,5 +1,6 @@
+import { LoadUnit } from '@eggjs/tegg-metadata';
 import { RootProtoManager } from './RootProtoManager';
 
 export interface ControllerRegister {
-  register(rootProtoManager: RootProtoManager): Promise<void>;
+  register(rootProtoManager: RootProtoManager, loadUnit?: LoadUnit): Promise<void>;
 }