Unified the injection traits