fixed small bug
attempted fix