Gradle script fix