From f9db26182c314683089eb5882f14671c8a2b9a1e Mon Sep 17 00:00:00 2001
From: Sergey Roytman <sergeyro@monday.com>
Date: Thu, 21 Mar 2024 14:05:27 +0200
Subject: [PATCH] refactoring

---
 packages/core/src/components/Button/Button.tsx         | 3 ++-
 packages/core/src/components/IconButton/IconButton.tsx | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/core/src/components/Button/Button.tsx b/packages/core/src/components/Button/Button.tsx
index f7354b2b11..11f109c25e 100644
--- a/packages/core/src/components/Button/Button.tsx
+++ b/packages/core/src/components/Button/Button.tsx
@@ -51,6 +51,7 @@ export interface ButtonProps extends VibeComponentProps {
   successText?: string;
   /** loading boolean which switches the text to a loader */
   loading?: boolean;
+  /** className which is applied to loader container **/
   loaderClassName?: string;
   style?: React.CSSProperties;
   /** displays the active state */
@@ -348,7 +349,7 @@ const Button: VibeComponent<ButtonProps, unknown> & {
       return (
         <button {...buttonProps} key={`${id}-loading`}>
           <span className={cx(styles.loader, loaderClassName)}>
-            <Loader className={cx(styles.loaderSvg)} />
+            <Loader className={styles.loaderSvg} />
             <span aria-hidden className={styles.textPlaceholder}>
               {buttonContent}
             </span>
diff --git a/packages/core/src/components/IconButton/IconButton.tsx b/packages/core/src/components/IconButton/IconButton.tsx
index f45b23ab0f..9db3cf594d 100644
--- a/packages/core/src/components/IconButton/IconButton.tsx
+++ b/packages/core/src/components/IconButton/IconButton.tsx
@@ -12,8 +12,8 @@ import { ComponentDefaultTestId } from "../../tests/constants";
 import { backwardCompatibilityForProperties } from "../../helpers/backwardCompatibilityForProperties";
 import Button from "../Button/Button";
 import { BUTTON_ICON_SIZE, ButtonColor, ButtonType } from "../Button/ButtonConstants";
-import styles from "./IconButton.module.scss";
 import { getStyle } from "../../helpers/typesciptCssModulesHelper";
+import styles from "./IconButton.module.scss";
 
 export interface IconButtonProps extends VibeComponentProps {
   /**
@@ -87,6 +87,7 @@ export interface IconButtonProps extends VibeComponentProps {
   insetFocus?: boolean;
   /** Specifies the tab order of an element */
   tabIndex?: number;
+  /** loading boolean which switches the text to a loader */
   loading?: boolean;
 }