fixing the small bug