Fix color issue of `p code`