From 29020c1200f1d496a86f4cda9a5249da1c499074 Mon Sep 17 00:00:00 2001
From: Nicolas Chaulet <nicolas.chaulet@elastic.co>
Date: Wed, 7 Jul 2021 16:37:18 -0400
Subject: [PATCH] [Fleet] Fix add agent in the package policy table (#104749)

---
 .../components/package_policy_actions_menu.tsx       | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/x-pack/plugins/fleet/public/components/package_policy_actions_menu.tsx b/x-pack/plugins/fleet/public/components/package_policy_actions_menu.tsx
index 9743135d5f1c..7b0a300ac9dc 100644
--- a/x-pack/plugins/fleet/public/components/package_policy_actions_menu.tsx
+++ b/x-pack/plugins/fleet/public/components/package_policy_actions_menu.tsx
@@ -28,6 +28,7 @@ export const PackagePolicyActionsMenu: React.FunctionComponent<{
   const { getHref } = useLink();
   const hasWriteCapabilities = useCapabilities().write;
   const refreshAgentPolicy = useAgentPolicyRefresh();
+  const [isActionsMenuOpen, setIsActionsMenuOpen] = useState(false);
 
   const onEnrollmentFlyoutClose = useMemo(() => {
     return () => setIsEnrollmentFlyoutOpen(false);
@@ -48,7 +49,10 @@ export const PackagePolicyActionsMenu: React.FunctionComponent<{
     // </EuiContextMenuItem>,
     <EuiContextMenuItem
       icon="plusInCircle"
-      onClick={() => setIsEnrollmentFlyoutOpen(true)}
+      onClick={() => {
+        setIsActionsMenuOpen(false);
+        setIsEnrollmentFlyoutOpen(true);
+      }}
       key="addAgent"
     >
       <FormattedMessage
@@ -112,7 +116,11 @@ export const PackagePolicyActionsMenu: React.FunctionComponent<{
           />
         </EuiPortal>
       )}
-      <ContextMenuActions items={menuItems} />
+      <ContextMenuActions
+        isOpen={isActionsMenuOpen}
+        items={menuItems}
+        onChange={(isOpen) => setIsActionsMenuOpen(isOpen)}
+      />
     </>
   );
 };