diff --git a/src/main/java/run/halo/migrate/MigratePlugin.java b/src/main/java/run/halo/migrate/MigratePlugin.java
index 5266e7b..15d80b0 100644
--- a/src/main/java/run/halo/migrate/MigratePlugin.java
+++ b/src/main/java/run/halo/migrate/MigratePlugin.java
@@ -1,8 +1,8 @@
 package run.halo.migrate;
 
-import org.pf4j.PluginWrapper;
 import org.springframework.stereotype.Component;
 import run.halo.app.plugin.BasePlugin;
+import run.halo.app.plugin.PluginContext;
 
 /**
  * @author guqing
@@ -10,15 +10,15 @@
  */
 @Component
 public class MigratePlugin extends BasePlugin {
-    
-    public MigratePlugin(PluginWrapper wrapper) {
-        super(wrapper);
+
+    public MigratePlugin(PluginContext pluginContext) {
+        super(pluginContext);
     }
-    
+
     @Override
     public void start() {
     }
-    
+
     @Override
     public void stop() {
     }