Also, why don't you do the same for RandomForest? I really don't see the point of this change :s