[MINOR] Fix gradle build