fixed small bug

fixed small bug