diff --git a/src/index.d.ts b/src/index.d.ts
index 1cd9ec4..62f918d 100644
--- a/src/index.d.ts
+++ b/src/index.d.ts
@@ -267,7 +267,8 @@ export declare const tv: TV;
 
 export declare const defaultConfig: TVConfig;
 
-export type VariantProps<Component extends (...args: any) => any> = Omit<
-  OmitUndefined<Parameters<Component>[0]>,
-  "class" | "className"
->;
+export type VariantProps<T> = T extends {variants: infer V}
+  ? V extends TVVariantsDefault<any, undefined>
+    ? {[K in TVVariantKeys<V, any>[number]]?: keyof V[K]}
+    : never
+  : never;