fixes small bug