diff --git a/src/main/java/aquality/selenium/browser/devtools/JavaScriptHandling.java b/src/main/java/aquality/selenium/browser/devtools/JavaScriptHandling.java index 17e7151..d758b86 100644 --- a/src/main/java/aquality/selenium/browser/devtools/JavaScriptHandling.java +++ b/src/main/java/aquality/selenium/browser/devtools/JavaScriptHandling.java @@ -12,6 +12,7 @@ import org.openqa.selenium.devtools.idealized.Events; import org.openqa.selenium.devtools.idealized.Javascript; import org.openqa.selenium.devtools.idealized.ScriptId; +import org.openqa.selenium.devtools.idealized.target.model.SessionID; import org.openqa.selenium.devtools.v85.page.Page; import org.openqa.selenium.devtools.v85.page.model.ScriptIdentifier; import org.openqa.selenium.devtools.v85.runtime.Runtime; @@ -111,7 +112,9 @@ private void removeInitializationScriptCore(InitializationScript script) { final Field pinnedScripts = Javascript.class.getDeclaredField("pinnedScripts"); pinnedScripts.setAccessible(true); //noinspection unchecked - ((Map)pinnedScripts.get(engine)).remove(script.getScriptSource()); + ((Map>)pinnedScripts.get(engine)) + .get(tools.getDevToolsSession().getCdpSession()) + .remove(script.getScriptSource()); pinnedScripts.setAccessible(false); } catch (ReflectiveOperationException e) { AqualityServices.getLogger().fatal("Error while removing initialization script", e);