Fix gradle build problem