Fix problem running on travis