more types

PostHog · Oct 2, 2024 · d12c958 · d12c958
1 parent e1dda95
commit d12c958
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/frontend/src/scenes/insights/InsightPageHeader.tsx b/frontend/src/scenes/insights/InsightPageHeader.tsx
@@ -38,7 +38,7 @@ import { userLogic } from 'scenes/userLogic'
 
 import { tagsModel } from '~/models/tagsModel'
 import { DataTableNode, NodeKind } from '~/queries/schema'
-import { isDataTableNode, isDataVisualizationNode, isHogQLQuery } from '~/queries/utils'
+import { isDataTableNode, isDataVisualizationNode, isEventsQuery, isHogQLQuery } from '~/queries/utils'
 import {
     ExporterFormat,
     InsightLogicProps,
@@ -90,8 +90,7 @@ export function InsightPageHeader({ insightLogicProps }: { insightLogicProps: In
     const [addToDashboardModalOpen, setAddToDashboardModalOpenModal] = useState<boolean>(false)
 
     const showCohortButton =
-        isHogQLQuery(query) ||
-        ((isDataTableNode(query) || isDataVisualizationNode(query)) && isHogQLQuery(query.source))
+        isDataTableNode(query) || isDataVisualizationNode(query) || isHogQLQuery(query) || isEventsQuery(query)
 
     return (
         <>