fix unit test for travis ci