Switched to setting labels in model classes