Revert "removed colorization fix"

This reverts commit 9e7672fe1b163986046ce70daf04e99d341d25d1.