def after_step(context, step):
    time.sleep(2)
    if step.status == "failed":
        allure.attach(context.browser.driver.get_screenshot_as_file
                      ('screenshots\{}.png'.format
                       (step.name)),
        name="Screenshot",
        attachment_type=AttachmentType.PNG)