Fix DI mistake