update highlight on removal also