diff --git a/tests/python/operators/fermi_hubbard_test.py b/tests/python/operators/fermi_hubbard_test.py index 69a74f6fd..9b6cb1805 100644 --- a/tests/python/operators/fermi_hubbard_test.py +++ b/tests/python/operators/fermi_hubbard_test.py @@ -16,8 +16,7 @@ import scipy import ffsim -from ffsim.operators.fermi_hubbard import fermi_hubbard_1d, fermi_hubbard_2d -from ffsim.operators.fermion_action import cre_a, cre_b, des_a, des_b +from ffsim import cre_a, cre_b, des_a, des_b, fermi_hubbard_1d, fermi_hubbard_2d def test_fermi_hubbard_1d():