Fix editor buttons not working