storybook-addon-material-ui: Fails with HMR

It works fine till you change a file and hot reloading kicks in:

It seems the context get lost and the injected theme vanished. Versions:

    "@storybook/addons": "3.2.0",
    "@storybook/react": "3.2.5",
    "storybook-addon-material-ui": "0.8.0",