repaired a small bug