fixed small bug, looking into another