Removed block in restore_icon_state