Update the tests for the last change.