Fix bug with updating checkmark