added small bug