UI: Fix unmount components on Canvas/Docs tab switch (#8625)

UI: Fix unmount components on Canvas/Docs tab switch
storybookjs · Nov 9, 2019 · b430c32 · b430c32
1 parent c4ba569
commit b430c32
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/core/src/client/preview/start.js b/lib/core/src/client/preview/start.js
@@ -200,12 +200,14 @@ export default function start(render, { decorateStory } = {}) {
       case 'docs':
         if (previousKind != null && (kind !== previousKind || viewMode !== previousViewMode)) {
           storyStore.cleanHooksForKind(previousKind);
+          ReactDOM.unmountComponentAtNode(document.getElementById('docs-root'));
         }
         break;
       case 'story':
       default:
         if (previousId != null && (id !== previousId || viewMode !== previousViewMode)) {
           storyStore.cleanHooks(previousId);
+          ReactDOM.unmountComponentAtNode(document.getElementById('root'));
         }
     }
 
@@ -224,7 +226,6 @@ export default function start(render, { decorateStory } = {}) {
         default: {
           if (previousViewMode === 'docs') {
             document.getElementById('docs-root').setAttribute('hidden', true);
-            ReactDOM.unmountComponentAtNode(document.getElementById('docs-root'));
             document.getElementById('root').removeAttribute('hidden');
           }
         }