Fix the bug for color