clean up of the DI