Fix the test on travis