diff --git a/analyzer/python/ikos/highlight.py b/analyzer/python/ikos/highlight.py
index 2726c327..486dc9fb 100644
--- a/analyzer/python/ikos/highlight.py
+++ b/analyzer/python/ikos/highlight.py
@@ -56,14 +56,14 @@ def __init__(self):
     def get_style_defs(self, cl):
         return ''
 
-    def wrap(self, src, out):
+    def wrap(self, src):
         pass
 
 
 def dummy_highlight(data, lexer, formatter):
     lines = ((1, html.escape(line)) for line in data.split('\n'))
     code = ''
-    for _, line in formatter.wrap(lines, None):
+    for _, line in formatter.wrap(lines):
         code += line
     return code
 
diff --git a/analyzer/python/ikos/view.py b/analyzer/python/ikos/view.py
index 4e9ed5d6..721df195 100644
--- a/analyzer/python/ikos/view.py
+++ b/analyzer/python/ikos/view.py
@@ -422,7 +422,7 @@ def __init__(self, file):
         self.call_contexts = {}
         self.checks = {}
 
-    def wrap(self, source, outfile):
+    def wrap(self, source):
         return self._wrap_code(source)
 
     def _wrap_code(self, source):