Fix gradle issue