I thought we should return `*RandomForest` here?