Fix little update issue