Fix problem with test on travis