AgGrid jumping to specific page/row

I am currently having the aggrid table with pagination and editable columns in my local.
Once I edit the content, the table refreshes back to first page even though the edit was made on some other page. Leading to the bad user experience.
I am currently on streamlit version 1.36.

Any workaround on this topic would be helpful.

Thank you in advance!