Fix gradle build issue