Fixed a couple errors in the tests.