Fix icon class for new version