diff --git a/kittens/diff/main.py b/kittens/diff/main.py
index c8b7711b4..488977145 100644
--- a/kittens/diff/main.py
+++ b/kittens/diff/main.py
@@ -32,7 +32,7 @@ from .search import BadRegex, Search
 try:
     from .highlight import initialize_highlighter, highlight_collection
 except ImportError:
-    initialize_highlighter = None
+    initialize_highlighter = highlight_collection = None
 
 
 INITIALIZING, COLLECTED, DIFFED, COMMAND, MESSAGE = range(5)