Keeping track of last updatemenus button pressed

I have a row of four buttons to control which traces are visible in a plotly map. This is done based on this example:

Is there some way I can keep track of which button is currently active?

For example, trace C is selected as shown below.

and then when a state on the map is clicked, the map is redrawn at a county level view zoomed into the state.

How can I maintain that trace C is still selected when the map is redrawn? Currently the map resets to trace A.