Fixing error with colors

See #65 for issue