From 2b4a6113451460eff85246f4c677e30d39398c91 Mon Sep 17 00:00:00 2001 From: Juanjo Alvarez Date: Tue, 7 Jan 2025 17:41:34 +0100 Subject: [PATCH] fix Signed-off-by: Juanjo Alvarez --- ddtrace/appsec/_iast/taint_sinks/header_injection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ddtrace/appsec/_iast/taint_sinks/header_injection.py b/ddtrace/appsec/_iast/taint_sinks/header_injection.py index 46ec7c8fa85..6f0a0a2efdc 100644 --- a/ddtrace/appsec/_iast/taint_sinks/header_injection.py +++ b/ddtrace/appsec/_iast/taint_sinks/header_injection.py @@ -92,7 +92,7 @@ def unpatch(): try_unwrap("django.http.response", "HttpResponseBase.__setitem__") try_unwrap("django.http.response", "ResponseHeaders.__setitem__") try_unwrap("starlette.datastructures", "MutableHeaders.__setitem__") - try_unwrap("starlette.responses", "response.init_headers") + try_unwrap("starlette.responses", "Response.init_headers") set_module_unpatched("flask", default_attr="_datadog_header_injection_patch") set_module_unpatched("django", default_attr="_datadog_header_injection_patch")