Fix storybook

[chiraggshah]

Mohit had mentioned that it was fixed in #632 but doesn't seem like it.
@paulsojan _a : Can you check if you can run storybook locally? And if not, then we can fix it.