Game Paused while widget open, unpause loses focus

Lets say for example my Inventory is open, and i press pause. The game pauses just fine and un-pauses just fine, however when i return to game after un-pausing the inventory window that HAD focus, it now has lost focus.

How do i retain what was last focus? Do i store what I had focus as a variable? im lost on this…

Solved. Store the active button as a variable in the player controller and on unpause i set focus to that variable.