diff --git a/vcr/stubs/__init__.py b/vcr/stubs/__init__.py index 5f7fac7..c1c4654 100644 --- a/vcr/stubs/__init__.py +++ b/vcr/stubs/__init__.py @@ -47,8 +47,9 @@ def parse_headers(header_list): def serialize_headers(response): + headers = response.headers if response.msg is None else response.msg out = {} - for key, values in compat.get_headers(response.msg): + for key, values in compat.get_headers(headers): out.setdefault(key, []) out[key].extend(values) return out