diff --git a/src/rubrix/server/tasks/text_classification/metrics.py b/src/rubrix/server/tasks/text_classification/metrics.py
index f97690d0e6..af7f4406fc 100644
--- a/src/rubrix/server/tasks/text_classification/metrics.py
+++ b/src/rubrix/server/tasks/text_classification/metrics.py
@@ -4,7 +4,7 @@
 from sklearn.metrics import precision_recall_fscore_support
 from sklearn.preprocessing import MultiLabelBinarizer
 
-from rubrix.server.tasks.commons.metrics import CommonTasksMetrics
+from rubrix.server.tasks.commons.metrics import CommonTasksMetrics, GenericRecord
 from rubrix.server.tasks.commons.metrics.model.base import (
     BaseMetric,
     PythonMetric,