diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh
index 85619f881f4b..76536ce0abe0 100755
--- a/ci/docker/runtime_functions.sh
+++ b/ci/docker/runtime_functions.sh
@@ -326,7 +326,8 @@ java_package_integration_test() {
     apt-get update -y
     apt-get install gradle -y
     # build java prokect
-    ./gradle build -x javadoc
+    cd /work/mxnet/java-package
+    ./gradlew build -x javadoc
     # generate native library
     ./gradlew :native:buildLocalLibraryJarDefault
     ./gradlew :native:mkl-linuxJar
diff --git a/java-package/scripts/setup_gradle_ubuntu.sh b/java-package/scripts/setup_gradle_ubuntu.sh
deleted file mode 100644
index 5450430fa6d5..000000000000
--- a/java-package/scripts/setup_gradle_ubuntu.sh
+++ /dev/null
@@ -1,15 +0,0 @@
-integration_test() {
-    # install gradle
-    add-apt-repository ppa:cwchien/gradle
-    apt-get update -y
-    apt-get install gradle -y
-    # build java prokect
-    cd /work/mxnet/java-package
-    ./gradle build -x javadoc
-    # generate native library
-    ./gradlew :native:buildLocalLibraryJarDefault
-    ./gradlew :native:mkl-linuxJar
-    # run integration
-    ./gradlew :integration:run
-}
-