fix: data viz node doesn't persist display type (#26164)

PostHog · Nov 14, 2024 · bc59a1e · bc59a1e
1 parent aafaea1
commit bc59a1e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/frontend/src/queries/nodes/DataVisualization/dataVisualizationLogic.ts b/frontend/src/queries/nodes/DataVisualization/dataVisualizationLogic.ts
@@ -283,6 +283,12 @@ export const dataVisualizationLogic = kea<dataVisualizationLogicType>([
         setConditionalFormattingRulesPanelActiveKeys: (keys: string[]) => ({ keys }),
     })),
     reducers(({ props }) => ({
+        query: [
+            props.query,
+            {
+                setQuery: (_, { node }) => node,
+            },
+        ],
         visualizationType: [
             ChartDisplayType.ActionsTable as ChartDisplayType,
             {
@@ -553,7 +559,6 @@ export const dataVisualizationLogic = kea<dataVisualizationLogicType>([
                 return columns.filter((n) => n.type.isNumerical)
             },
         ],
-        query: [(_state, props) => [props.query], (query) => query],
         showEditingUI: [
             (state, props) => [state.insightMode, props.insightLogicProps],
             (insightMode, insightLogicProps) => {