minor update

fix a small bug