Fix travis after migrating to pytest