Fixed a unit-test error