removed a small bug