fix an import for new extension structure

RadioAstronomySoftwareGroup · Jun 25, 2024 · ee5f918 · ee5f918
1 parent 589d32a
commit ee5f918
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/uvbeam/test_uvbeam.py b/tests/uvbeam/test_uvbeam.py
@@ -17,9 +17,10 @@
 from astropy.coordinates import Angle
 from astropy.io import fits
 
-from pyuvdata import UVBeam, _uvbeam, utils
+from pyuvdata import UVBeam, utils
 from pyuvdata.data import DATA_PATH
 from pyuvdata.testing import check_warnings
+from pyuvdata.uvbeam import _uvbeam
 
 from .test_cst_beam import cst_files, cst_yaml_file
 from .test_mwa_beam import filename as mwa_beam_file