Selenium 4: Cómo capturar una captura de pantalla

Necesito un método After para capturar una captura de pantalla compatible con Selenium 4 Java

Solución del problema

@After
public void tearDown(Scenario scenario) {
 if (scenario.isFailed()) {
 final byte[] screenshot = ((TakesScreenshot) Driver.get()).getScreenshotAs(OutputType.BYTES);
 scenario.attach(screenshot, "image/png", "screenshot");
 }
 Driver.closeDriver();
}