diff --git a/dte_adj/base.py b/dte_adj/base.py
index 4650cc0..cfa6605 100644
--- a/dte_adj/base.py
+++ b/dte_adj/base.py
@@ -424,7 +424,7 @@ def predict(
 
         Args:
             treatment_arm (int): The index of the treatment arm.
-            outcomes (np.ndarray): Scalar values to be used for computing the cumulative distribution.
+            locations (np.ndarray): Scalar values to be used for computing the cumulative distribution.
             display_progress (bool, optional): Whether to display a progress bar. Defaults to True.
 
         Returns: