small bug solved