diff --git a/package.json b/package.json
index f6399f5..3c850e9 100644
--- a/package.json
+++ b/package.json
@@ -33,7 +33,9 @@
     "analyze": "size-limit --why",
     "release": "yarn build && auto shipit"
   },
-  "peerDependencies": {},
+  "peerDependencies": {
+    "react": "^16.8.0 || ^17.0.0"
+  },
   "husky": {
     "hooks": {
       "pre-commit": "tsdx lint"
diff --git a/src/types.ts b/src/types.ts
index d584ede..85814c6 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -1,6 +1,8 @@
 import { ArgTypes, Parameters, BaseDecorators } from '@storybook/addons';
 import type { Story } from '@storybook/react';
-import { StoryFnReactReturnType } from '@storybook/react/dist/client/preview/types';
+import { ReactElement } from 'react';
+
+type StoryFnReactReturnType = ReactElement<unknown>;
 
 /**
  * Object representing the preview.ts module