fixed small bug
update code