small bug that needed to be fixed