diff --git a/test/run_java_unit_tests.js b/test/run_java_unit_tests.js
index f26641c1..a4c9c165 100644
--- a/test/run_java_unit_tests.js
+++ b/test/run_java_unit_tests.js
@@ -26,7 +26,7 @@ var superspawn = require('cordova-common').superspawn;
 // First we make sure the gradlew helper file is built and ready.
 var GradleBuilder = require('../bin/templates/cordova/lib/builders/GradleBuilder');
 var builder = new GradleBuilder(__dirname);
-var needs_gradlew_built = builder.runGradleWrapper('gradle', 'build.gradle');
+var needs_gradlew_built = builder.runGradleWrapper('gradle');
 
 if (!needs_gradlew_built) {
     // Due to interface of gradle builder, if the gradlew file already exists, `runGradleWrapper` returns undefined.
diff --git a/test/wrapper.gradle b/test/wrapper.gradle
new file mode 100644
index 00000000..11514912
--- /dev/null
+++ b/test/wrapper.gradle
@@ -0,0 +1,3 @@
+wrapper {
+    gradleVersion = '4.4.1'
+}