Review feedback

deephaven · Mar 6, 2025 · f3c2d06 · f3c2d06
1 parent d0e6dec
commit f3c2d06
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/web/client-api/src/main/java/io/deephaven/web/client/api/tree/JsTreeTable.java b/web/client-api/src/main/java/io/deephaven/web/client/api/tree/JsTreeTable.java
@@ -219,17 +219,17 @@ public JsTreeTable(WorkerConnection workerConnection, JsWidget widget) {
                     constituentColumns.put(column.getName(), column);
                 } else {
                     columns[columns.length] = column;
-                    if (definition.isRollupGroupByColumn() && !definition.isRollupConstituentNodeColumn()) {
+                    if (definition.isRollupGroupByColumn()) {
                         groupedColumns.push(column);
 
                         if (hasConstituentColumns) {
                             column.setConstituentType(columnDefsByName.get(true).get(definition.getName()).getType());
                         }
                     }
-                    if (hasConstituentColumns && definition.getRollupAggregationInputColumn() != null
-                            && !definition.getRollupAggregationInputColumn().isEmpty()) {
+                    String aggInputCol = definition.getRollupAggregationInputColumn();
+                    if (hasConstituentColumns && aggInputCol != null && !aggInputCol.isEmpty()) {
                         column.setConstituentType(
-                                columnDefsByName.get(true).get(definition.getRollupAggregationInputColumn()).getType());
+                                columnDefsByName.get(true).get(aggInputCol).getType());
                     }
                 }
             }