one more unit test fix