Small bug fixed in update dataset