Update tests to reflect change.