Update the expected exception from DI