diff --git a/src/amrex/space1d/__init__.pyi b/src/amrex/space1d/__init__.pyi
index dad62af4..de2f2bc8 100644
--- a/src/amrex/space1d/__init__.pyi
+++ b/src/amrex/space1d/__init__.pyi
@@ -37,11 +37,12 @@ from __future__ import annotations
 
 import os as os
 
-from amrex.Array4 import register_Array4_extension
-from amrex.ArrayOfStructs import register_AoS_extension
-from amrex.MultiFab import register_MultiFab_extension
-from amrex.ParticleContainer import register_ParticleContainer_extension
-from amrex.PODVector import register_PODVector_extension
+from amrex.extensions.Array4 import register_Array4_extension
+from amrex.extensions.ArrayOfStructs import register_AoS_extension
+from amrex.extensions.MultiFab import register_MultiFab_extension
+from amrex.extensions.ParticleContainer import register_ParticleContainer_extension
+from amrex.extensions.PODVector import register_PODVector_extension
+from amrex.extensions.StructOfArrays import register_SoA_extension
 from amrex.space1d.amrex_1d_pybind import (
     AlmostEqual,
     AMReX,
@@ -253,7 +254,6 @@ from amrex.space1d.amrex_1d_pybind import (
     unpack_ids,
     write_single_level_plotfile,
 )
-from amrex.StructOfArrays import register_SoA_extension
 
 from . import amrex_1d_pybind
 
diff --git a/src/amrex/space2d/__init__.pyi b/src/amrex/space2d/__init__.pyi
index f9b59c2f..c1f07cf6 100644
--- a/src/amrex/space2d/__init__.pyi
+++ b/src/amrex/space2d/__init__.pyi
@@ -37,11 +37,12 @@ from __future__ import annotations
 
 import os as os
 
-from amrex.Array4 import register_Array4_extension
-from amrex.ArrayOfStructs import register_AoS_extension
-from amrex.MultiFab import register_MultiFab_extension
-from amrex.ParticleContainer import register_ParticleContainer_extension
-from amrex.PODVector import register_PODVector_extension
+from amrex.extensions.Array4 import register_Array4_extension
+from amrex.extensions.ArrayOfStructs import register_AoS_extension
+from amrex.extensions.MultiFab import register_MultiFab_extension
+from amrex.extensions.ParticleContainer import register_ParticleContainer_extension
+from amrex.extensions.PODVector import register_PODVector_extension
+from amrex.extensions.StructOfArrays import register_SoA_extension
 from amrex.space2d.amrex_2d_pybind import (
     AlmostEqual,
     AMReX,
@@ -277,7 +278,6 @@ from amrex.space2d.amrex_2d_pybind import (
     unpack_ids,
     write_single_level_plotfile,
 )
-from amrex.StructOfArrays import register_SoA_extension
 
 from . import amrex_2d_pybind
 
diff --git a/src/amrex/space3d/__init__.pyi b/src/amrex/space3d/__init__.pyi
index 5b282fcf..89fdc511 100644
--- a/src/amrex/space3d/__init__.pyi
+++ b/src/amrex/space3d/__init__.pyi
@@ -37,11 +37,12 @@ from __future__ import annotations
 
 import os as os
 
-from amrex.Array4 import register_Array4_extension
-from amrex.ArrayOfStructs import register_AoS_extension
-from amrex.MultiFab import register_MultiFab_extension
-from amrex.ParticleContainer import register_ParticleContainer_extension
-from amrex.PODVector import register_PODVector_extension
+from amrex.extensions.Array4 import register_Array4_extension
+from amrex.extensions.ArrayOfStructs import register_AoS_extension
+from amrex.extensions.MultiFab import register_MultiFab_extension
+from amrex.extensions.ParticleContainer import register_ParticleContainer_extension
+from amrex.extensions.PODVector import register_PODVector_extension
+from amrex.extensions.StructOfArrays import register_SoA_extension
 from amrex.space3d.amrex_3d_pybind import (
     AlmostEqual,
     AMReX,
@@ -253,7 +254,6 @@ from amrex.space3d.amrex_3d_pybind import (
     unpack_ids,
     write_single_level_plotfile,
 )
-from amrex.StructOfArrays import register_SoA_extension
 
 from . import amrex_3d_pybind