Tree structure using oj-row-expander on table, cursor pointing to the above selected.

Summary:

Hey Experts,

I have a tree structured table, using oj-row-expander having 3 levels in each node. The query is say i expand a node in the top and then edit a value and then go to a below node to expand it but the focus comes back to the the place i had selected the record before. how can I make it such that when expanding the focus comes to the place where we need to expand.

Attaching pic for reference.

Red Box is the task im trying to expand and yellow circle is where the screen takes it's focus to even after scrolling to the bottom task.