Fix test on travis