Correct update bug