small unit test fix