fixed small bug: