fix unit test helper