Fix armor-resetting bug