Correct error and update test.