diff --git a/code_comments/comment.py b/code_comments/comment.py
index 8850db9..f64e67b 100644
--- a/code_comments/comment.py
+++ b/code_comments/comment.py
@@ -6,7 +6,7 @@
 import re
 
 import trac.wiki.formatter
-from trac.mimeview.api import Context
+from trac.web.chrome import web_context
 from time import strftime, localtime
 from code_comments import db
 from trac.util import Markup
@@ -173,5 +173,5 @@ def default(self, o):
 def format_to_html(req, env, text):
     req = Mock(href=Href('/'), abs_href=Href('http://www.example.com/'),
                authname='anonymous', perm=MockPerm(), args={})
-    context = Context.from_request(req)
+    context = web_context(req)
     return trac.wiki.formatter.format_to_html(env, context, text)