fixing DI