diff --git a/storybook/stories/Delegation.stories.js b/storybook/stories/Delegation.stories.js
index f1a397e6c4..cb54fc1a5c 100644
--- a/storybook/stories/Delegation.stories.js
+++ b/storybook/stories/Delegation.stories.js
@@ -2,18 +2,34 @@
import React from 'react';
import { storiesOf } from '@storybook/react';
import { withKnobs, number } from '@storybook/addon-knobs';
+import StoryLayout from './support/StoryLayout';
+import StoryProvider from './support/StoryProvider';
import StoryDecorator from './support/StoryDecorator';
+import { CATEGORIES_BY_NAME } from '../../source/renderer/app/config/sidebarConfig';
import Delegation from '../../source/renderer/app/components/staking/Delegation';
const timeLeft = (78 * 60 + 10) * 60 * 1000;
-storiesOf('Delegation', module)
- .addDecorator((story, context) => (
-