Fix test fail on travis