Fix travis fails