diff --git a/node-schemas/build.gradle b/node-schemas/build.gradle
index 872980ce1a..bf5b2edabc 100644
--- a/node-schemas/build.gradle
+++ b/node-schemas/build.gradle
@@ -1,9 +1,14 @@
 apply plugin: 'kotlin'
 apply plugin: 'kotlin-kapt'
+apply plugin: 'idea'
 apply plugin: 'net.corda.plugins.publish-utils'
 
 description 'Corda node database schemas'
 
+buildscript {
+    ext.generatedSourcesDir = "${projectDir}/generated/source/kapt/main"
+}
+
 repositories {
     mavenLocal()
     mavenCentral()
@@ -26,7 +31,7 @@ sourceSets {
     }
     generated {
         java {
-            srcDir "${projectDir}/generated/source/kapt/main/"
+            srcDir generatedSourcesDir
             compileClasspath += sourceSets.main.compileClasspath + sourceSets.main.output
         }
     }
@@ -76,3 +81,9 @@ project.afterEvaluate {
 jar {
     from sourceSets.generated.output
 }
+
+idea {
+    module {
+        sourceDirs += project.sourceSets.generated.allSource
+    }
+}
\ No newline at end of file