diff --git a/src/non_local_detector/models/base.py b/src/non_local_detector/models/base.py
index 4f7d9d1..f15ed25 100644
--- a/src/non_local_detector/models/base.py
+++ b/src/non_local_detector/models/base.py
@@ -2211,6 +2211,7 @@ def predict(
             marginal_log_likelihood,
             _,
             _,
+            _,
         ) = self._predict(
             time=time,
             log_likelihood_args=(