Revert "removed colorization fix"

This reverts commit 025eba01260c902047c98cd45a183dd34d4684c3.