diff --git a/dan/predict/prediction.py b/dan/predict/prediction.py
index c6c000a7aac3c694479d42b87a8ef0d387777249..6c03c47df475bf37f7b8aa8a78bc29190b7408b9 100644
--- a/dan/predict/prediction.py
+++ b/dan/predict/prediction.py
@@ -115,7 +115,6 @@ class DAN:
         input_sizes,
         confidences=False,
         attentions=False,
-        confidences_sep=None,
     ):
         """
         Run prediction on an input image.