Fix Travis tests