fix class update issue