Fix the color wrap function
Merge pull request #186 from cattode/wrap-patch

Fix the color wrap function