di contracts cleanup