diff --git a/types/react-albus/index.d.ts b/types/react-albus/index.d.ts
index 48dfcb9a6b..be546011d4 100644
--- a/types/react-albus/index.d.ts
+++ b/types/react-albus/index.d.ts
@@ -28,9 +28,9 @@ export interface WizardComponentProps {
     wizard: WizardContext;
 }
 
-export const withWizard: <P>(
+export function withWizard<P>(
     component: React.ComponentType<P & WizardComponentProps>
-) => React.ComponentType<P>;
+): React.ComponentType<P>;
 
 export interface WizardProps {
     onNext?: (wizard: WizardContext) => void;
diff --git a/types/react-albus/react-albus-tests.tsx b/types/react-albus/react-albus-tests.tsx
index 5bdb3d13c4..2f31733b1d 100644
--- a/types/react-albus/react-albus-tests.tsx
+++ b/types/react-albus/react-albus-tests.tsx
@@ -1,6 +1,5 @@
 import * as React from "react";
 import { Wizard, Step, Steps, withWizard } from "react-albus";
-import { ButtonHTMLAttributes } from "react";
 
 const Example = () => (
     <Wizard