diff --git a/extended-it/build.gradle b/extended-it/build.gradle
index c16dd4e6e3..0f78343ca0 100644
--- a/extended-it/build.gradle
+++ b/extended-it/build.gradle
@@ -28,6 +28,7 @@ dependencies {
     }
 
     implementation project(":extended")
+    testImplementation project(':extended').sourceSets.main.allJava
 
     testImplementation group: 'us.fatehi', name: 'schemacrawler-mysql', version: '16.20.8'
     testImplementation group: 'org.postgresql', name: 'postgresql', version: '42.1.4'
diff --git a/extended/build.gradle b/extended/build.gradle
index 87504eb534..18868c8f67 100644
--- a/extended/build.gradle
+++ b/extended/build.gradle
@@ -52,7 +52,9 @@ task gitSubmoduleLoad {
 }
 
 dependencies {
-
+    // to fix unresolved imports with IntelliJ from version 2024.1.x
+    compileOnly sourceSets.main.java
+    
     apt project(':processor')
     apt group: 'org.neo4j', name: 'neo4j', version: neo4jVersionEffective   // mandatory to run @ServiceProvider based META-INF code generation