diff --git a/gradle/src/main/groovy/com/antfortune/freeline/FreelineInitializer.groovy b/gradle/src/main/groovy/com/antfortune/freeline/FreelineInitializer.groovy
index e5962c8..b36bd0b 100644
--- a/gradle/src/main/groovy/com/antfortune/freeline/FreelineInitializer.groovy
+++ b/gradle/src/main/groovy/com/antfortune/freeline/FreelineInitializer.groovy
@@ -207,7 +207,7 @@ class FreelineInitializer {
 
         if (buildScript == null || buildScript == '') {
             // set default build script
-            def isRootModuleTheMainModule = project.name == projectDescription.main_project_name
+            def isRootModuleTheMainModule = project.rootProject.name == projectDescription.main_project_name
             projectDescription.build_script = FreelineGenerator.generateBuildScript(isRootModuleTheMainModule, projectDescription.main_project_name as String, productFlavor)
         }