Fix for latest pytest release