Fixed a unit test problem