Honestly, I don't think he will remain a TM player for much longer. I still prefer Ezatolahi-Hajsafi in the middle DM positions. Ali Karimi is also a good option.
I mean, at the end of the day Ezatolahi missing the first game of the World Cup does open up a slot for someone taking that position. But I think it'll be Karimi. He's much more imposing and calm on the ball.