diff --git a/src/sensai/evaluation/evaluator.py b/src/sensai/evaluation/evaluator.py
index f2c9f9b78..a648d5cfd 100644
--- a/src/sensai/evaluation/evaluator.py
+++ b/src/sensai/evaluation/evaluator.py
@@ -415,7 +415,7 @@ class VectorClassificationModelEvaluator(VectorModelEvaluator[VectorClassificati
     def __init__(self,
             data: Optional[InputOutputData],
             test_data: InputOutputData = None,
-            params: VectorClassificationModelEvaluatorParams = None):
+            params: ClassificationEvaluatorParams = None):
         """
         Constructs an evaluator with test and training data.