update for small bug