Fix bug on travis