Fixed a couple tests.