diff --git a/tests/test_decision_trees/test_decision_trees.py b/tests/test_decision_trees/test_decision_trees.py
index 0d0bf0bd..c53387de 100644
--- a/tests/test_decision_trees/test_decision_trees.py
+++ b/tests/test_decision_trees/test_decision_trees.py
@@ -3,9 +3,12 @@
 
 from decision_tree.decision_tree_regression import decision_tree_regression
 from decision_tree.decision_tree_classification import decision_tree_classification
+from extra_tree.extra_tree_regression import extra_tree_regression
+from extra_tree.extra_tree_classification import extra_tree_classification
+
 DECISION_TREES = {
     "DECISION_TREE": [decision_tree_regression, decision_tree_classification],
-
+    "EXTRA TREE": [extra_tree_regression, extra_tree_classification],
 }
 
 @pytest.fixture(scope="session", autouse=True)