fixed the small bug that was