1 files changed, 7 insertions, 8 deletions
diff --git a/demo/plugins/plugin1/src/main/java/org/pf4j/demo/welcome/WelcomePlugin.java b/demo/plugins/plugin1/src/main/java/org/pf4j/demo/welcome/WelcomePlugin.java
index a0cecab..3064902 100644
--- a/demo/plugins/plugin1/src/main/java/org/pf4j/demo/welcome/WelcomePlugin.java
+++ b/demo/plugins/plugin1/src/main/java/org/pf4j/demo/welcome/WelcomePlugin.java
@@ -16,27 +16,26 @@
 package org.pf4j.demo.welcome;
 
 import org.apache.commons.lang.StringUtils;
-
-import org.pf4j.PluginWrapper;
+import org.pf4j.Extension;
 import org.pf4j.RuntimeMode;
+import org.pf4j.demo.api.DemoPlugin;
 import org.pf4j.demo.api.Greeting;
-import org.pf4j.Extension;
-import org.pf4j.Plugin;
+import org.pf4j.demo.api.PluginContext;
 
 /**
  * @author Decebal Suiu
  */
-public class WelcomePlugin extends Plugin {
+public class WelcomePlugin extends DemoPlugin {
 
-    public WelcomePlugin(PluginWrapper wrapper) {
-        super(wrapper);
+    public WelcomePlugin(PluginContext context) {
+        super(context);
     }
 
     @Override
     public void start() {
         log.info("WelcomePlugin.start()");
         // for testing the development mode
-        if (RuntimeMode.DEVELOPMENT.equals(wrapper.getRuntimeMode())) {
+        if (RuntimeMode.DEVELOPMENT.equals(context.getRuntimeMode())) {
             log.info(StringUtils.upperCase("WelcomePlugin"));
         }
     }