fixing small bug