minor update

minor update, bug fix