fixed small bug.