Fixed test for Travis-CI