Fix update bug