Fix button colour sticking, by using a selector