Fix a couple tests.