fix two lines of a small bug