diff --git a/mjx/mujoco/mjx/__init__.py b/mjx/mujoco/mjx/__init__.py
index 8656e618..d3e048d2 100644
--- a/mjx/mujoco/mjx/__init__.py
+++ b/mjx/mujoco/mjx/__init__.py
@@ -16,6 +16,7 @@
 
 # pylint:disable=g-importing-member
 from mujoco.mjx._src.collision_driver import collision
+from mujoco.mjx._src.collision_driver import ncon
 from mujoco.mjx._src.constraint import count_constraints
 from mujoco.mjx._src.constraint import make_constraint
 from mujoco.mjx._src.device import device_get_into
@@ -43,7 +44,7 @@ from mujoco.mjx._src.smooth import kinematics
 from mujoco.mjx._src.smooth import rne
 from mujoco.mjx._src.smooth import transmission
 from mujoco.mjx._src.solver import solve
-from mujoco.mjx._src.support import is_sparse
 from mujoco.mjx._src.support import full_m
+from mujoco.mjx._src.support import is_sparse
 from mujoco.mjx._src.support import mul_m
 from mujoco.mjx._src.types import *