Fixed a unit test error