Update tests to test new code.