Make the existing tests work with the new code