diff --git a/src/P2/processing.py b/src/P2/processing.py
index 44fffaf..cb64fa2 100644
--- a/src/P2/processing.py
+++ b/src/P2/processing.py
@@ -41,7 +41,7 @@ def predict_data(data, model, results, sample):
         df=results,
         model=model_name,
         prediction=prediction,
-        clusters=len(prediction),
+        clusters=len(set(prediction)),
         calinski=calinski,
         silhouette=silhouette,
         time=execution_time,