diff --git a/src/AggregateFunctions/AggregateFunctionDistinct.h b/src/AggregateFunctions/AggregateFunctionDistinct.h
index b481e2a28e77a5ad45a123c47ce2dd933e0a0d2d..b587bbebf6e26e6b9e656b7bf09b3981ba65938f 100644
--- a/src/AggregateFunctions/AggregateFunctionDistinct.h
+++ b/src/AggregateFunctions/AggregateFunctionDistinct.h
@@ -236,6 +236,11 @@ public:
         return true;
     }
 
+    bool isState() const override
+    {
+        return nested_func->isState();
+    }
+
     AggregateFunctionPtr getNestedFunction() const override { return nested_func; }
 };
 
diff --git a/tests/queries/0_stateless/01259_combinator_distinct_distributed.reference b/tests/queries/0_stateless/01259_combinator_distinct_distributed.reference
index 096d5703292eeaa05bff31fd708d019c76c4dbf6..72a41ac1d8433d2c0ac0c9ed16a1432f40238bc1 100644
--- a/tests/queries/0_stateless/01259_combinator_distinct_distributed.reference
+++ b/tests/queries/0_stateless/01259_combinator_distinct_distributed.reference
@@ -2,3 +2,7 @@
 [0,1,2,3,4,5,6,7,8,9,10,11,12]
 20
 0.49237
+78
+[0,1,2,3,4,5,6,7,8,9,10,11,12]
+20
+0.49237
diff --git a/tests/queries/0_stateless/01259_combinator_distinct_distributed.sql b/tests/queries/0_stateless/01259_combinator_distinct_distributed.sql
index f851e64dbcbce6a8d78706c1eb85e3a00dbe0b4f..f95d2d87b8ef4d6e09693c8acee97977bc10bca7 100644
--- a/tests/queries/0_stateless/01259_combinator_distinct_distributed.sql
+++ b/tests/queries/0_stateless/01259_combinator_distinct_distributed.sql
@@ -1,3 +1,12 @@
+SET distributed_aggregation_memory_efficient = 1;
+
+SELECT sum(DISTINCT number % 13) FROM remote('127.0.0.{1,2}', numbers_mt(100000));
+SELECT arraySort(groupArray(DISTINCT number % 13)) FROM remote('127.0.0.{1,2}', numbers_mt(100000));
+SELECT finalizeAggregation(countState(DISTINCT toString(number % 20))) FROM remote('127.0.0.{1,2}', numbers_mt(100000));
+SELECT round(corrStable(DISTINCT x, y), 5) FROM (SELECT number % 10 AS x, number % 5 AS y FROM remote('127.0.0.{1,2}', numbers(1000)));
+
+SET distributed_aggregation_memory_efficient = 0;
+
 SELECT sum(DISTINCT number % 13) FROM remote('127.0.0.{1,2}', numbers_mt(100000));
 SELECT arraySort(groupArray(DISTINCT number % 13)) FROM remote('127.0.0.{1,2}', numbers_mt(100000));
 SELECT finalizeAggregation(countState(DISTINCT toString(number % 20))) FROM remote('127.0.0.{1,2}', numbers_mt(100000));