diff --git a/springwolf-core/src/test/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/AbstractChannelScannerTest.java b/springwolf-core/src/test/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/AbstractChannelScannerTest.java
index ca2557c14..e68c7e8a9 100644
--- a/springwolf-core/src/test/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/AbstractChannelScannerTest.java
+++ b/springwolf-core/src/test/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/AbstractChannelScannerTest.java
@@ -7,6 +7,7 @@
 import io.github.stavshamir.springwolf.asyncapi.scanners.components.ComponentsScanner;
 import io.github.stavshamir.springwolf.asyncapi.types.channel.operation.message.Message;
 import io.github.stavshamir.springwolf.asyncapi.types.channel.operation.message.PayloadReference;
+import io.github.stavshamir.springwolf.configuration.AsyncApiDocket;
 import io.github.stavshamir.springwolf.schemas.DefaultSchemasService;
 import lombok.Data;
 import lombok.NoArgsConstructor;
@@ -39,8 +40,12 @@ public class AbstractChannelScannerTest {
     @MockBean
     private ComponentsScanner componentsScanner;
 
+    @MockBean
+    private AsyncApiDocket docket;
+
     private void setClassToScan(Class<?> classToScan) {
         Set<Class<?>> classesToScan = singleton(classToScan);
+        when(docket.getBasePackage()).thenReturn("io.github.stavshamir.springwolf.asyncapi.scanners.channels");
         when(componentsScanner.scanForComponents(anyString())).thenReturn(classesToScan);
     }
 
diff --git a/springwolf-core/src/test/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/TestChannelScanner.java b/springwolf-core/src/test/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/TestChannelScanner.java
index 6b2f4b917..9195b1b9a 100644
--- a/springwolf-core/src/test/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/TestChannelScanner.java
+++ b/springwolf-core/src/test/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/TestChannelScanner.java
@@ -9,11 +9,6 @@
 
 public class TestChannelScanner extends AbstractChannelScanner<AbstractChannelScannerTest.TestChannelListener> {
 
-    @Override
-    protected String getBasePackage() {
-        return "io.github.stavshamir.springwolf.asyncapi.scanners.channels";
-    }
-
     @Override
     protected Class<AbstractChannelScannerTest.TestChannelListener> getListenerAnnotationClass() {
         return AbstractChannelScannerTest.TestChannelListener.class;