Fixed a bug with live update