Fixed a unit test.