Fix travis testing