Try fix unit test