diff --git a/py/server/tests/test_rollup_tree_table.py b/py/server/tests/test_rollup_tree_table.py
index 2f1d76bdafd..409a9319f30 100644
--- a/py/server/tests/test_rollup_tree_table.py
+++ b/py/server/tests/test_rollup_tree_table.py
@@ -4,7 +4,7 @@
 import unittest
 
 from deephaven import read_csv, empty_table
-from deephaven.agg import sum_, avg, count_, first, last, max_, min_, std, abs_sum, \
+from deephaven.agg import sum_, avg, count_, count_where, first, last, max_, min_, std, abs_sum, \
     var
 from deephaven.filters import Filter
 from deephaven.table import NodeType
@@ -18,6 +18,7 @@ def setUp(self):
         self.aggs_for_rollup = [
             avg(["aggAvg=var"]),
             count_("aggCount"),
+            count_where("aggCountWhere", "var > 0"),
             first(["aggFirst=var"]),
             last(["aggLast=var"]),
             max_(["aggMax=var"]),