Potential fix to TriggerTrait.