One more unit test fix