Fix travis tests (hopefully)