diff --git a/privacy/dp_query/quantile_adaptive_clip_sum_query.py b/privacy/dp_query/quantile_adaptive_clip_sum_query.py
index 3391eb1..a8eb400 100644
--- a/privacy/dp_query/quantile_adaptive_clip_sum_query.py
+++ b/privacy/dp_query/quantile_adaptive_clip_sum_query.py
@@ -120,7 +120,6 @@ class QuantileAdaptiveClipSumQuery(dp_query.DPQuery):
         self._sum_query.initial_global_state(),
         self._clipped_fraction_query.initial_global_state())
 
-  @tf.function
   def derive_sample_params(self, global_state):
     """See base class."""
     gs = global_state