DI Exception restructure