Fix colorization unittest

Fix the unit test for colorization. That previously was not running.
Can we delete this file