Fixed test for travis ci.