fixed small bug
small bug