diff --git a/src/mock_vws/_flask_server/vwq.py b/src/mock_vws/_flask_server/vwq.py index 063374059..838c4a7bc 100644 --- a/src/mock_vws/_flask_server/vwq.py +++ b/src/mock_vws/_flask_server/vwq.py @@ -10,9 +10,9 @@ from http import HTTPStatus import requests +import werkzeug from flask import Flask, Response, request from pydantic import BaseSettings -from werkzeug.datastructures import Headers from mock_vws._query_tools import ( ActiveMatchingTargetsDeleteProcessing, @@ -105,7 +105,7 @@ def handle_exceptions(exc: ValidatorException) -> Response: headers=exc.headers, ) - response.headers = Headers(exc.headers) + response.headers = werkzeug.datastructures.Headers(exc.headers) return response diff --git a/src/mock_vws/_flask_server/vws.py b/src/mock_vws/_flask_server/vws.py index d5a21679a..1fc1e7f99 100644 --- a/src/mock_vws/_flask_server/vws.py +++ b/src/mock_vws/_flask_server/vws.py @@ -14,9 +14,9 @@ from http import HTTPStatus import requests +import werkzeug from flask import Flask, Response, request from pydantic import BaseSettings -from werkzeug.datastructures import Headers from mock_vws._constants import ResultCodes, TargetStatuses from mock_vws._database_matchers import get_database_matching_server_keys @@ -134,7 +134,7 @@ def handle_exceptions(exc: ValidatorException) -> Response: headers=exc.headers, ) - response.headers = Headers(exc.headers) + response.headers = werkzeug.datastructures.Headers(exc.headers) return response