fixed small bug, oops