Fix the test response expectations