diff --git a/gradle/apiGenerator.gradle b/gradle/apiGenerator.gradle
index 254f837..6991f5b 100644
--- a/gradle/apiGenerator.gradle
+++ b/gradle/apiGenerator.gradle
@@ -1,3 +1,5 @@
+final File generatedModelsDirectory = new File("${project.buildDir}/generated/source/models")
+
 repositories {
     maven {
         url 'https://dl.bintray.com/touchin/touchin-tools'
@@ -17,12 +19,10 @@ dependencies {
 
 android {
     sourceSets {
-        main.java.srcDirs += "build/generated/source/models"
+        main.java.srcDirs += generatedModelsDirectory
     }
 }
 
-final File generatedModelsDirectory = new File("${project.buildDir}/generated/source/models")
-
 def generateJsonModelsTask = tasks.create("apiGenerator") doLast {
     javaexec {
         main = "-jar"