Fix live update bug