diff --git a/tests/test_bms_multi_model_output.py b/tests/test_bms_multi_model_output.py
index 7ea5d36ac..3d3962800 100644
--- a/tests/test_bms_multi_model_output.py
+++ b/tests/test_bms_multi_model_output.py
@@ -1,8 +1,8 @@
 import numpy as np
 import pytest
-from theorist.bms import Tree
 
 from autora.skl.bms import BMSRegressor
+from autora.theorist.bms import Tree
 
 
 @pytest.fixture