Fixing switching stories

eliperelman · May 3, 2017 · 57f32d8 · 57f32d8
1 parent ee8f6cb
commit 57f32d8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/Stories.js b/src/Stories.js
@@ -50,8 +50,10 @@ export class Stories extends React.Component {
     };
   }
 
-  selectStory({ Component, states }) {
-    this.setState({ Component, props: states.values().next().value });
+  selectStory({ kind }) {
+    const { Component, states } = this.state.components.get(kind);
+
+    this.setState({ Component, props: states.values().next().value, kind });
   }
 
   selectProps(props) {