Minor fix with colors