Fix a bug with color output