Fix colorization unittest

Fix the unit test for colorization. That previously was not running.
What about the other one: documentRegistry?