[IMP] fix the color issue