fixed small bug
minor fixed