fix unit test... again...