Fix bug for Travis