Adapt to new tempest-test package