hot fix unit test