fixed small bug2