1 files changed, 5 insertions, 0 deletions
diff --git a/spark-velocity2/src/main/java/me/lucko/spark/velocity/Velocity2SparkPlugin.java b/spark-velocity2/src/main/java/me/lucko/spark/velocity/Velocity2SparkPlugin.java
index a788a59..0ea4bf8 100644
--- a/spark-velocity2/src/main/java/me/lucko/spark/velocity/Velocity2SparkPlugin.java
+++ b/spark-velocity2/src/main/java/me/lucko/spark/velocity/Velocity2SparkPlugin.java
@@ -110,6 +110,11 @@ public class Velocity2SparkPlugin implements SparkPlugin, SimpleCommand {
     }
 
     @Override
+    public ClassSourceLookup createClassSourceLookup() {
+        return new Velocity2ClassSourceLookup(this.proxy.pluginManager());
+    }
+
+    @Override
     public PlatformInfo getPlatformInfo() {
         return new Velocity2PlatformInfo(this.proxy);
     }