fixed one small bug