fixing small bug: