diff --git a/apps/frontend/src/presentation/components/ui/button.jsx b/apps/frontend/src/presentation/components/ui/button.jsx
index 3d8094b..2aec148 100644
--- a/apps/frontend/src/presentation/components/ui/button.jsx
+++ b/apps/frontend/src/presentation/components/ui/button.jsx
@@ -1,9 +1,23 @@
-import React, { useState } from "react";
+import React from "react";
+import PropTypes from "prop-types";
 import { Button as AntButton } from "antd";
 
 // Bug fix: ButtonWrapper now forwards refs correctly
-export const Button = React.forwardRef((props, ref) => {
-  return <AntButton htmlType="button" ref={ref} style={{ boxShadow: 'none' }} {...props} />;
+const Button = React.forwardRef(({ style, ...props }, ref) => {
+  return (
+    <AntButton
+      htmlType="button"
+      ref={ref}
+      style={{ boxShadow: "none", ...style }}
+      {...props}
+    />
+  );
 });
 
+Button.displayName = "Button";
 
+Button.propTypes = {
+  style: PropTypes.object,
+};
+
+export { Button };