diff --git a/pyproject.toml b/pyproject.toml
index 7e1ede86..1d8ba5f1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -54,4 +54,4 @@ extend-select = [
 "!src/**.py" = ["D"]
 
 [tool.uv.sources]
-openarm-mujoco = { git = "https://github.com/enactic/openarm_mujoco.git", subdirectory = "v2", rev = "cba9759f7245ae3925f91c6600bdccc130cbf4e3" }
+openarm-mujoco = { git = "https://github.com/enactic/openarm_mujoco.git" }
diff --git a/src/openarm_control/config.py b/src/openarm_control/config.py
index 3f44b03b..9c4e8a81 100644
--- a/src/openarm_control/config.py
+++ b/src/openarm_control/config.py
@@ -20,8 +20,8 @@
 
 import mujoco
 import numpy as np
-import openarm_mujoco_v2 as openarm_mujoco
-from openarm_mujoco_v2 import JointResolver
+import openarm_mujoco.v2 as openarm_mujoco
+from openarm_mujoco.v2 import JointResolver
 
 _DEFAULT_XML = openarm_mujoco.openarm_cell_xml()