diff --git a/judge/highlight_code.py b/judge/highlight_code.py
index 979eebf..5d9add3 100644
--- a/judge/highlight_code.py
+++ b/judge/highlight_code.py
@@ -27,9 +27,9 @@ else:
         if linenos:
             return mark_safe(
                 pygments.highlight(
-                    code, lexer, HtmlCodeFormatter(cssclass=cssclass, linenos="table", wrapcode=True)
+                    code, lexer, pygments.formatters.HtmlFormatter(cssclass=cssclass, linenos="table", wrapcode=True)
                 )
             )
         return mark_safe(
-            pygments.highlight(code, lexer, HtmlCodeFormatter(cssclass=cssclass, wrapcode=True))
+            pygments.highlight(code, lexer, pygments.formatters.HtmlFormatter(cssclass=cssclass, wrapcode=True))
         )