DI clean up for