Fix travis CI tests