fix base on demo document