Возможно ли прикрепить скриншот страницы к каждому "then" степу (pytest-bdd, selenium, allure)

Я пишу тесты на pytest-bdd + selenium + allure. И столкнулся с необходимостью делать скрин страницы на каждый шаг проверки. В pytest-bdd есть хук "after step" и в моем варианте он выглядит так:

def pytest_bdd_after_step(request, feature, scenario, step, step_func, step_func_args):
    if step.keyword == 'Then':
        driver = request.getfixturevalue('driver')
        allure.attach(driver.get_screenshot_as_png(), name='screenshot',
                      attachment_type=allure.attachment_type.PNG)

Это работает, но скрины крепятся в репорте не к конкретному шагу. а после всех шагов:

Собственно, вопрос в следующем: существует ли возможность прикрепить скриншот к степу "then" в хуке "after"? Спасибо!