diff --git a/apis/python/tests/test_experiment_query.py b/apis/python/tests/test_experiment_query.py
index d308cd15ae..b7ad58d961 100644
--- a/apis/python/tests/test_experiment_query.py
+++ b/apis/python/tests/test_experiment_query.py
@@ -14,10 +14,10 @@
 from somacore import AxisQuery, options
 
 import tiledbsoma as soma
-from tiledbsoma import SOMATileDBContext, _factory, pytiledbsoma
+from tiledbsoma import SOMATileDBContext, _factory, pytiledbsoma, ExperimentAxisQuery
 from tiledbsoma._collection import CollectionBase
 from tiledbsoma._experiment import Experiment
-from tiledbsoma._query import Axis, ExperimentAxisQuery
+from tiledbsoma._query import Axis
 from tiledbsoma.experiment_query import X_as_series
 
 from tests._util import raises_no_typeguard