small bug change