Update for brush issue