Fix icon updating