fix type

liam-hq · Dec 16, 2024 · 4cef3ec · 4cef3ec
1 parent 19e499a
commit 4cef3ec
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/...ages/erd-core/src/components/ERDRenderer/ERDContent/Toolbar/TidyUpButton/TidyUpButton.tsx b/...ages/erd-core/src/components/ERDRenderer/ERDContent/Toolbar/TidyUpButton/TidyUpButton.tsx
@@ -1,17 +1,20 @@
 import { IconButton, TidyUpIcon } from '@liam-hq/ui'
 import { ToolbarButton } from '@radix-ui/react-toolbar'
-import type { FC } from 'react'
+import { type FC, useCallback } from 'react'
 import { useAutoLayout } from '../../useAutoLayout'
 
 export const TidyUpButton: FC = () => {
   const { handleLayout } = useAutoLayout()
+  const handleClick = useCallback(() => {
+    handleLayout()
+  }, [handleLayout])
 
   return (
     <ToolbarButton asChild>
       <IconButton
         icon={<TidyUpIcon />}
         tooltipContent="Tidy up"
-        onClick={handleLayout}
+        onClick={handleClick}
       />
     </ToolbarButton>
   )