From 82013ed065532f81940be31e19cad72cfdf12a13 Mon Sep 17 00:00:00 2001
From: Kamil Gabryjelski <kamil.gabryjelski@gmail.com>
Date: Fri, 7 Apr 2023 16:07:18 +0200
Subject: [PATCH] Fix test

---
 .../Chart/ChartContextMenu/useContextMenu.test.tsx          | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/superset-frontend/src/components/Chart/ChartContextMenu/useContextMenu.test.tsx b/superset-frontend/src/components/Chart/ChartContextMenu/useContextMenu.test.tsx
index e46b700da0815..ebab36b14fe75 100644
--- a/superset-frontend/src/components/Chart/ChartContextMenu/useContextMenu.test.tsx
+++ b/superset-frontend/src/components/Chart/ChartContextMenu/useContextMenu.test.tsx
@@ -36,8 +36,12 @@ global.featureFlags = {
 
 const setup = ({
   onSelection = noOp,
-  displayedItems = [ContextMenuItem.All],
+  displayedItems = ContextMenuItem.All,
   additionalConfig = {},
+}: {
+  onSelection?: () => void;
+  displayedItems?: ContextMenuItem | ContextMenuItem[];
+  additionalConfig?: Record<string, any>;
 } = {}) => {
   const { result } = renderHook(() =>
     useContextMenu(