Update the expected exception from the DI container

@suhasj