I'd rather it be Lee even if he dies because he's had no development for a very long time.
All of post time skip is like this. I'm fine with the show sticking with main characters, but the only teens to really do anything of note in the Shippuden era were Naruto, Sasuke, Shikamaru, Gaara, and Kabuto. Nobody else has done jack shit that lasted for like more than 1 episode/chapter.
Neji showed up just to get killed. Rock Lee has literally done nothing in nearly 300 chapters/episodes. Hinata got one chapter/episode. Tenten got a new weapon and we didn't even get to see her use it. Temari threw a wind attack at Madara. Ino stood in the background twice. Choji got an episode/chapter.
Rather than a bunch of boring one vs one fights with people like Kankuro and the Black Dude from the Raikage's place, I would have preferred this war arc have like Neji's team have to tackle a bad guy. Give Hinata's squad an Akatsuki member to face off against back when they were relevant. Anything to get some of these characters back in the fray to show how they've grown post-time skip.