fix unit test (again)