diff --git a/src/frontend/src/controllers/API/queries/flows/use-patch-update-flow.ts b/src/frontend/src/controllers/API/queries/flows/use-patch-update-flow.ts
index e818319c3..af5280546 100644
--- a/src/frontend/src/controllers/API/queries/flows/use-patch-update-flow.ts
+++ b/src/frontend/src/controllers/API/queries/flows/use-patch-update-flow.ts
@@ -12,7 +12,7 @@ interface IPatchUpdateFlow {
   description: string;
   folder_id: string | null | undefined;
   endpoint_name: string | null | undefined;
-  locked: boolean | null | undefined;
+  locked?: boolean | null | undefined;
 }
 
 export const usePatchUpdateFlow: useMutationFunctionType<
diff --git a/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx
index 7796e21fd..5eeef286c 100644
--- a/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx
+++ b/src/frontend/src/pages/FlowPage/components/nodeToolbarComponent/index.tsx
@@ -542,13 +542,13 @@ export default function NodeToolbarComponent({
           >
             <SelectTrigger className="w-62">
               <ShadTooltip content="Show More" side="top">
-                <div>
+                <div data-testid="more-options-modal">
                   <Button
                     className="node-toolbar-buttons h-[2rem] w-[2rem]"
                     variant="ghost"
                     onClick={handleButtonClick}
                     size="node-toolbar"
-                    data-testid="more-options-modal"
+                    asChild
                   >
                     <IconComponent name="MoreHorizontal" className="h-4 w-4" />
                   </Button>