udpated a small bug