diff --git a/gradle/src/main/groovy/com/antfortune/freeline/FreelineInitializer.groovy b/gradle/src/main/groovy/com/antfortune/freeline/FreelineInitializer.groovy
index 0449b5d..3aff1af 100644
--- a/gradle/src/main/groovy/com/antfortune/freeline/FreelineInitializer.groovy
+++ b/gradle/src/main/groovy/com/antfortune/freeline/FreelineInitializer.groovy
@@ -199,7 +199,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)
         }