Getting the correct pytest version