Fix colorization unittest

Fix the unit test for colorization. That previously was not running.