Would it make sense to introduce variables that determine icon color for the different types of button so this can be removed from here? Feel free to create a separate follow-up ticket for that task.