benchmark: update for the changes in for_each module