Update with correct labels