Gradle build fix