Armour upgrades resets

I've come across a bug with armour upgrades.  As soon as i remove the armour set all the upgrades disappear. Re-equipping does not resolve this.

Upgrading the armour further improves the stats, but the locked feature slots remain locked.