He was definitely playing more restricted role under Sarri, I think Sarri was trying to protect him by not allowing to him move around as much and thus keep midfield defensively in balance.
He has also improved quite a bit, especially physically. Just shows that players coming from other leagues need their time to adapt to the requirements of Premier League.