Button display fixes