fix the unit coloration issue regarding custom weapons.