Set layout buttons to highlight correctly when selected