Forgot a commit. This updates the tests.