Fixes

Author: wojtekmaj

src/index.tsx

@@ -18,10 +18,10 @@ const STATES = {
   INIT: 'init',
   PENDING: 'pending',
   SUCCESS: 'success',
-};
+} as const;
 
 const AsyncButton = React.forwardRef(
-  <T extends React.ElementType>(
+  <T extends React.ElementType = 'button'>(
     {
       as,
       errorConfig,
@@ -33,7 +33,7 @@ const AsyncButton = React.forwardRef(
     }: AsyncButtonProps<T>,
     ref: React.ForwardedRef<any>,
   ) => {
-    const [buttonState, setButtonState] = useState(STATES.INIT);
+    const [buttonState, setButtonState] = useState<typeof STATES[keyof typeof STATES]>(STATES.INIT);
     const cancellablePromise = useRef<ReturnType<typeof makeCancellable>>();
     const timeout = useRef<ReturnType<typeof setTimeout>>();