added small bug fix