Fixed my unit test.