fix unit test for travis ci 2