Fix code color