diff --git a/build.gradle.kts b/build.gradle.kts
index 3dc0fa1fa54..caf7efe6bf4 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -58,7 +58,7 @@ cmaker {
 }
 
 val repo = jgit.repo()
-val commitCount = (repo?.commitCount("refs/remotes/origin/master") ?: 1) + 4200
+val commitCount = (repo?.commitCount("HEAD") ?: 1) + 4200
 val latestTag = repo?.latestTag?.removePrefix("v") ?: "1.0"
 
 val injectedPackageName by extra("com.android.shell")