Fix color output