diff --git a/node-schemas/build.gradle b/node-schemas/build.gradle
index 56e61cfd6c..df382cb787 100644
--- a/node-schemas/build.gradle
+++ b/node-schemas/build.gradle
@@ -27,8 +27,13 @@ sourceSets {
     generated {
         java {
             srcDir "${projectDir}/generated/source/kapt/main/"
+            compileClasspath += sourceSets.main.compileClasspath + sourceSets.main.output
         }
     }
+    test {
+        compileClasspath += sourceSets.generated.output + sourceSets.main.compileClasspath
+        runtimeClasspath += compileClasspath
+    }
 }
 
 dependencies {
@@ -60,8 +65,10 @@ project.afterEvaluate {
     jar.dependsOn.each { jar.dependsOn.remove it }
     buildGenerated.dependsOn.each { buildGenerated.dependsOn.remove it }
     jar.dependsOn buildGenerated
+    compileTestKotlin.dependsOn buildGenerated
     buildGenerated.dependsOn classes
     compileKotlin.dependsOn.remove kaptKotlin
+    compileTestKotlin.dependsOn.remove kaptTestKotlin
     buildKapt.dependsOn kaptKotlin
     buildKapt.mustRunAfter kaptKotlin
 }