* unit test fix