Fix UserActionTrait logic