diff --git a/lib/addons/src/types.ts b/lib/addons/src/types.ts
index e2efc16e9423..77b463d8627e 100644
--- a/lib/addons/src/types.ts
+++ b/lib/addons/src/types.ts
@@ -92,8 +92,13 @@ export interface OptionsParameter extends Object {
 export type StoryGetter = (context: StoryContext) => any;
 
 export type LegacyStoryFn<ReturnType = unknown> = (p?: StoryContext) => ReturnType;
-export type ArgsStoryFn<ReturnType = unknown> = (a?: Args, p?: StoryContext) => ReturnType;
-export type StoryFn<ReturnType = unknown> = LegacyStoryFn<ReturnType> | ArgsStoryFn<ReturnType>;
+export type ArgsStoryFn<ReturnType = unknown, ArgsType extends Args = Args> = (
+  a?: ArgsType,
+  p?: StoryContext
+) => ReturnType;
+export type StoryFn<ReturnType = unknown, ArgsType = Args> =
+  | LegacyStoryFn<ReturnType>
+  | ArgsStoryFn<ReturnType, ArgsType>;
 
 export type StoryWrapper = (
   getStory: StoryGetter,