correcting a small bug