fix unit test for travis ci 4