make sure buttons color correctly