Fix button coloring