Updating a couple of tests.