Make the wrapStory decorator simple to use.

example/stories/index.js

@@ -12,7 +12,7 @@ import {
 import Button from './Button';
 
 storiesOf('Button', module)
-  .addDecorator((story, context) => (withKnobs(story)(context)))
+  .addDecorator(withKnobs)
   .add('default view', () => (
     <Button
       onClick={ action('button clicked') }

src/KnobManager.js

@@ -30,8 +30,7 @@ export default class KnobManager {
     return knobStore.get(name).value;
   }
 
-  wrapStory(storyFn, channel) {
+  wrapStory(channel, storyFn, context) {
-    return context => {
     const key = `${context.kind}:::${context.story}`;
     let knobStore = this.knobStoreMap[key];
 
@@ -42,6 +41,5 @@ export default class KnobManager {
 
     const props = { context, storyFn, channel, knobStore };
     return (<WrapStory {...props} />);
-    };
   }
 }

src/index.js

@@ -23,7 +23,7 @@ export function object(name, value) {
   return manager.knob(name, { type: 'object', value });
 }
 
-export function withKnobs(storyFn) {
+export function withKnobs(storyFn, context) {
   const channel = addons.getChannel();
-  return manager.wrapStory(storyFn, channel);
+  return manager.wrapStory(channel, storyFn, context);
 }