1 files changed, 8 insertions, 5 deletions
diff --git a/examples/modify_response_body.py b/examples/modify_response_body.py
index 03dfeaa4..b4632248 100644
--- a/examples/modify_response_body.py
+++ b/examples/modify_response_body.py
@@ -3,16 +3,19 @@
 import sys
 
 
-def start(context):
+state = {}
+
+
+def start():
     if len(sys.argv) != 3:
         raise ValueError('Usage: -s "modify_response_body.py old new"')
     # You may want to use Python's argparse for more sophisticated argument
     # parsing.
-    context.old, context.new = sys.argv[1], sys.argv[2]
+    state["old"], state["new"] = sys.argv[1].encode(), sys.argv[2].encode()
 
 
-def response(context, flow):
+def response(flow):
     flow.response.content = flow.response.content.replace(
-        context.old,
-        context.new
+        state["old"],
+        state["new"]
     )