Fixed a UI issue involving traits in units in the Unit Store sometimes not being accurately reflected.