Fix the test_forest.py