diff --git a/nyoka/skl/skl_to_pmml.py b/nyoka/skl/skl_to_pmml.py
index b701670..4746105 100644
--- a/nyoka/skl/skl_to_pmml.py
+++ b/nyoka/skl/skl_to_pmml.py
@@ -1370,6 +1370,8 @@ def get_inner_segments(model, derived_col_names, col_names, index):
                 recordCount=estm.tree_.n_node_samples[0]
             )
         mining_fields = list()
+        if model.__class__.__name__ in ['RandomForestClassifier','RandomForestRegressor']:
+            col_names = derived_col_names
         for feat in col_names:
             mining_fields.append(pml.MiningField(name=feat))
         segments.append(