diff --git a/build.gradle b/build.gradle
index 8cb4079..89a2fb2 100644
--- a/build.gradle
+++ b/build.gradle
@@ -84,20 +84,14 @@ tasks.named('test') {
 }
 
 // querydsl
-def generatedSrcDir = "$buildDir/generated/sources/annotationProcessor/java/main"
+def generatedSrcDir = layout.buildDirectory.dir("generated/sources/annotationProcessor/java/main").get().asFile
+
 clean {
 	delete file(generatedSrcDir)
 }
 tasks.withType(JavaCompile).configureEach {
 	options.annotationProcessorGeneratedSourcesDirectory = file(generatedSrcDir)
 }
-sourceSets {
-    main {
-        java {
-            srcDir generatedSrcDir
-        }
-    }
-}
 
 jib {
   from {