minor update & fix