diff --git a/web_app/src/components/ui/button.tsx b/web_app/src/components/ui/button.tsx
index f93111c..6647b5f 100644
--- a/web_app/src/components/ui/button.tsx
+++ b/web_app/src/components/ui/button.tsx
@@ -49,7 +49,7 @@ const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
       <Comp
         className={cn(
           buttonVariants({ variant, size, className }),
-          "outline-none cursor-default"
+          "outline-none cursor-default select-none"
         )}
         ref={ref}
         tabIndex={-1}
diff --git a/web_app/src/lib/states.ts b/web_app/src/lib/states.ts
index 327eae6..6a82131 100644
--- a/web_app/src/lib/states.ts
+++ b/web_app/src/lib/states.ts
@@ -493,7 +493,8 @@ export const useStore = createWithEqualityFn<AppState & AppAction>()(
           imageWidth,
           imageHeight,
           [maskLineGroup],
-          maskImages
+          maskImages,
+          BRUSH_COLOR
         )
         if (useLastLineGroup) {
           const temporaryMask = await canvasToImage(maskCanvas)