Fix DI handler in trait.