Fix issue when uncharging and the icon doesn't change