fix unit tests fail