diff --git a/mjx/mujoco/mjx/_src/collision_convex.py b/mjx/mujoco/mjx/_src/collision_convex.py
index 55d96ef498..ea9c62d9dc 100644
--- a/mjx/mujoco/mjx/_src/collision_convex.py
+++ b/mjx/mujoco/mjx/_src/collision_convex.py
@@ -14,9 +14,8 @@
 # ==============================================================================
 """Convex collisions."""
 
-from collections.abc import Callable
 import functools
-from typing import Tuple, Union
+from typing import Callable, Tuple, Union
 
 import jax
 from jax import numpy as jp