diff --git a/app/build.gradle b/app/build.gradle
index 3eb948bb67fa86ea4736636ad3c4b1e6ca6c2477..5133b8627d84d6aba81d9e454d5bf2f7bd27cfaf 100644
--- a/app/build.gradle
+++ b/app/build.gradle
@@ -52,13 +52,14 @@ android {
     insecure {
       dimension "implementation"
     }
-    dafualt {
+    defualt {
       dimension "mode"
     }
+    //Configurations for custom branded app.
     custom {
       dimension "mode"
       //Change the package name as needed
-      applicationId "org.sample.customs"
+      applicationId "org.sample.custom"
       //Set app name here
       appName = "Custom"
       resValue "string", "app_name", appName
@@ -370,25 +371,40 @@ def getCurrentFlavorForBetaOrRelease() {
   }
 }
 
-task checkApplicationIdForCustomFlavor (type: Exec ) {
-  def currFlavor = getCurrentFlavorForBetaOrRelease()
-  if(currFlavor.contains("custom")) {
+def getCurrentFlavor() {
+  Gradle gradle = getGradle()
+  String tskReqStr = gradle.getStartParameter().getTaskRequests().toString()
+
+  Pattern pattern;
+
+  if (tskReqStr.contains("assemble"))
+    pattern = Pattern.compile("assemble(\\w+)(Beta|Release|Debug)")
+  else
+    pattern = Pattern.compile("generate(\\w+)(Beta|Release|Debug)")
+
+  Matcher matcher = pattern.matcher(tskReqStr)
+
+  if (matcher.find())
+    return matcher.group(1).toLowerCase()
+  else {
+    return "";
+  }
+}
+
+task checkApplicationIdForCustomFlavor(type: Exec) {
+  def currFlavor = getCurrentFlavor()
+  if (currFlavor.contains("custom")) {
     android.applicationVariants.all { variant ->
       def mergedFlavor = variant.mergedFlavor
-      if (variant.flavorName.toString().equalsIgnoreCase(currFlavor)) {
-        if (mergedFlavor.getApplicationId().equalsIgnoreCase("org.sample.custom")) {
-          throw new ExecutionException("ERROR: please change the applicationId(org.sample.custom) if you want to build a custom branded app!")
-        }
-      }
+      if (variant.flavorName.toString().equalsIgnoreCase(currFlavor) &&
+              mergedFlavor.getApplicationId().equalsIgnoreCase("org.sample.custom"))
+        throw new ExecutionException("ERROR: please change the applicationId(org.sample.custom) if you want to build a custom branded app!")
     }
   }
 }
 
-task checkAppNameForCustomFlavor (type: Exec ) {
-  def currFlavor = getCurrentFlavorForBetaOrRelease()
-  if(currFlavor.contains("custom")) {
-    if (appName.equalsIgnoreCase("custom")) {
-      throw new ExecutionException("ERROR: please change the appName(Custom) if you want to build a custom branded app!")
-    }
-  }
+task checkAppNameForCustomFlavor(type: Exec) {
+  def currFlavor = getCurrentFlavor()
+  if (currFlavor.contains("custom") && appName.equalsIgnoreCase("custom"))
+    throw new ExecutionException("ERROR: please change the appName(Custom) if you want to build a custom branded app!")
 }
\ No newline at end of file