diff --git a/IssueLabeler/src/Trainer/Trainer.cs b/IssueLabeler/src/Trainer/Trainer.cs
index 8efc117..d74e694 100644
--- a/IssueLabeler/src/Trainer/Trainer.cs
+++ b/IssueLabeler/src/Trainer/Trainer.cs
@@ -89,12 +89,16 @@ static async Task CreateModel(string dataPath, string modelPath, ModelType type,
 
     await action.WriteStatusAsync("Building pipeline...");
 
+    string[] featureColumns = type == ModelType.Issue
+        ? ["Title", "Body"]
+        : ["Title", "Body", "FileNames", "FolderNames"];
+
     var xf = mlContext.Transforms;
     var pipeline = xf.Conversion.MapValueToKey(inputColumnName: "Label", outputColumnName: "LabelKey")
         .Append(xf.Text.FeaturizeText(
             "Features",
             new TextFeaturizingEstimator.Options(),
-            columns.Select(c => c.Name).ToArray()))
+            featureColumns))
         .AppendCacheCheckpoint(mlContext)
         .Append(mlContext.MulticlassClassification.Trainers.SdcaMaximumEntropy("LabelKey"))
         .Append(xf.Conversion.MapKeyToValue("PredictedLabel"));