Unresponsive lasso and box selectors on button activated traces for sankey diagram

Here is a sankey diagram containing data for two traces. The traces are button activated. I want to flip between traces and use the modebar box/lasso selectors to group nodes.

The lasso and box selectors are unresponsive on the second trace (selection does not work). However, the selectors work just fine on the first trace (selection works). To replicate the issue, click on a button before selecting a node.

What am I doing wrong? Thank you

import plotly
import plotly.offline as offline
import plotly.graph_objs as go

fig = go.Figure()

trace1 = go.Sankey(
    node = dict(
        label = ["carrot", "radish", "potato", "liccorice", "hamburger", "stomach", "soil"],
        color = ["blue", "red", "green", "yellow", "brown", "magenta", "green"]
    ),
    link = dict(
        source = [0,1,3,6,6],
        target = [5,5,0,1,3],
        value =  [3,4,3,4,3]
        ))

trace2 = trace1
data = [trace1,trace2]

updatemenus = list([
    dict(type="buttons",
         buttons = list([
             dict(label = 'Selection works',
                  method = 'update',
                  args = [{'visible':[True,False]},
                           {'title': 'Selecion works'}]),
             dict(label = 'Selection does not work',
                  method = 'update',
                  args = [{'visible':[False,True]},   
                           {'title': 'Selection does not work'}]),
])
)])


layout = go.Layout(showlegend=True,
              updatemenus=updatemenus)

fig = dict(data=data,layout=layout)

offline.plot(fig, filename='food.html')

three reasons why you should buy plotly pro: support open source, get great support, host your plots and dashboards online