fixing small bug.