diff --git a/build.gradle b/build.gradle
index 53eeeb62f60664a7d7b9371466ab6deaeb56aed9..15cf1b93cc8c04ab2e4e621fb55d4566fc21a421 100644
--- a/build.gradle
+++ b/build.gradle
@@ -48,6 +48,27 @@ dependencies {
   testRuntimeOnly "org.junit.jupiter:junit-jupiter-engine"
 }
 
+sourceSets {
+  main {
+    java {
+      srcDirs = ['src/main']
+    }
+    resources {
+      srcDirs = ['src/resources']
+      exclude '**/*.java'
+    }
+  }
+  test {
+    java {
+      srcDirs = ['test']
+    }
+    resources {
+      srcDirs = ['test']
+      exclude '**/*.java'
+    }
+  }
+}
+
 task faultSections(type: Sync) {
   from "../nshm-fault-sections"
   include "**/*.geojson"