diff --git a/build.gradle b/build.gradle
index be49100a..c4acdae4 100644
--- a/build.gradle
+++ b/build.gradle
@@ -376,7 +376,7 @@ configure(rootProject) {
 
         doLast() {
             def gradleOpts = "-XX:MaxPermSize=1024m -Xmx1024m"
-            def gradleBatOpts = "$gradleOpts -XX:MaxHeapSize=256"
+            def gradleBatOpts = "$gradleOpts -XX:MaxHeapSize=256m"
             File wrapperFile = file('gradlew')
             wrapperFile.text = wrapperFile.text.replace("DEFAULT_JVM_OPTS=",
                 "GRADLE_OPTS=\"$gradleOpts \$GRADLE_OPTS\"\nDEFAULT_JVM_OPTS=")