Remove unused import for test_tree