Fix bug for Travis (2)