diff --git a/flow360/_public_namespace.py b/flow360/_public_namespace.py
index 29f5e078c..b59715771 100644
--- a/flow360/_public_namespace.py
+++ b/flow360/_public_namespace.py
@@ -189,6 +189,7 @@
     ReferenceGeometry,
     SeedpointVolume,
     Sphere,
+    VoxelGrid,
 )
 from flow360.component.simulation.run_control.run_control import RunControl
 from flow360.component.simulation.run_control.stopping_criterion import (
@@ -347,6 +348,7 @@ def configure(apikey: str, environment: str = None, profile: str = "default") ->
     "PointArray",
     "AngleExpression",
     "Box",
+    "VoxelGrid",
     "GenericReferenceCondition",
     "TransitionModelSolver",
     "TurbulenceModelControls",
diff --git a/flow360/component/simulation/primitives.py b/flow360/component/simulation/primitives.py
index c067a4996..57cf00eb5 100644
--- a/flow360/component/simulation/primitives.py
+++ b/flow360/component/simulation/primitives.py
@@ -27,7 +27,10 @@
     _MirroredEntityBase,
     compute_bbox_tolerance,
 )
-from flow360_schema.models.entities.volume_entities import (
+
+# VoxelGrid lands in flow360_schema alongside the schema-side flex PR; drop
+# the inline pylint disable once the new flow360_schema release is pinned.
+from flow360_schema.models.entities.volume_entities import (  # pylint: disable=no-name-in-module
     AxisymmetricBody,
     AxisymmetricSegment,
     Box,
@@ -37,5 +40,6 @@
     GenericVolume,
     SeedpointVolume,
     Sphere,
+    VoxelGrid,
 )
 from flow360_schema.models.reference_geometry import ReferenceGeometry