Fix update issue