[FEATURE] Fix update bug fix