fixed small bug from previous update