make button colour work correctly