React to DI changes

The exception thrown by DI changed

@pranavkm