Fix bug in random forest example.