Fixes DI