removing small bug