fixed small bug created on modification