unit test fix!