Fix travis tests