Fix controller based on new custom field