diff --git a/bugbug/model.py b/bugbug/model.py
index 9ef60c51d3..fbae7595b7 100644
--- a/bugbug/model.py
+++ b/bugbug/model.py
@@ -370,7 +370,7 @@ def train(self, importance_cutoff=0.15, limit=None):
         X_gen, y = split_tuple_generator(lambda: self.items_gen(classes))
 
         # Extract features from the items.
-        X = self.extraction_pipeline.transform(X_gen)
+        X = self.extraction_pipeline.fit_transform(X_gen)
 
         # Calculate labels.
         y = np.array(y)