Fix the test for the error handling.