better screenshot for storybook.