Fix test running on travis