Fix Travis test