solution for color problem

@Wei789 please check this.
fix #1