diff --git a/tensorflow_privacy/__init__.py b/tensorflow_privacy/__init__.py
index dd1ca07..74e46df 100644
--- a/tensorflow_privacy/__init__.py
+++ b/tensorflow_privacy/__init__.py
@@ -32,6 +32,7 @@ else:
   from tensorflow_privacy.privacy.analysis.privacy_ledger import SampleEntry
 
   from tensorflow_privacy.privacy.dp_query.dp_query import DPQuery
+  from tensorflow_privacy.privacy.dp_query.dp_query import SumAggregationDPQuery
   from tensorflow_privacy.privacy.dp_query.gaussian_query import GaussianAverageQuery
   from tensorflow_privacy.privacy.dp_query.gaussian_query import GaussianSumQuery
   from tensorflow_privacy.privacy.dp_query.nested_query import NestedQuery