Fix test for travis ci.