updated a small bug