finally fix color issue of code