diff --git a/src/spikeinterface/curation/tests/test_train_manual_curation.py b/src/spikeinterface/curation/tests/test_train_manual_curation.py
index 0cb74d6d58..0a3ca2d45b 100644
--- a/src/spikeinterface/curation/tests/test_train_manual_curation.py
+++ b/src/spikeinterface/curation/tests/test_train_manual_curation.py
@@ -89,8 +89,6 @@ def test_apply_scaling_imputation(trainer):
     assert np.all(y_train == y_train_scaled)
     assert np.all(y_test == y_test_scaled)
 
-    print(type(scaler))
-
     assert isinstance(imputer, KNNImputer)
     assert isinstance(scaler, StandardScaler)
 
diff --git a/src/spikeinterface/curation/train_manual_curation.py b/src/spikeinterface/curation/train_manual_curation.py
index 23c45cf776..5f25d1e3e7 100644
--- a/src/spikeinterface/curation/train_manual_curation.py
+++ b/src/spikeinterface/curation/train_manual_curation.py
@@ -369,7 +369,7 @@ def process_test_data_for_classification(self):
             raise KeyError(f"{str(e)}, metrics_list contains invalid metric names")
 
         self.X = self.testing_metrics.reindex(columns=self.metric_names)
-        self.X = _format_metric_dataframe(self.testing_metrics)
+        self.X = _format_metric_dataframe(self.X)
 
     def apply_scaling_imputation(self, imputation_strategy, scaling_technique, X_train, X_test, y_train, y_test):
         """Impute and scale the data using the specified techniques."""