diff --git a/build.gradle b/build.gradle
index cdd5211c..eead570c 100644
--- a/build.gradle
+++ b/build.gradle
@@ -41,7 +41,7 @@ if (hasProperty("teamcity")) {
     version = teamcity["build.number"]
 
     // Fix for main branch being built
-    if (teamcity.build.branch.is_default) {
+    if (teamcity["teamcity.build.branch.is_default"]) {
         version = version.replaceAll("/", "-")
     }
 }