Fixed a unit test