diff --git a/.readthedocs.yml b/.readthedocs.yml
index 464475c6..e892543a 100644
--- a/.readthedocs.yml
+++ b/.readthedocs.yml
@@ -14,6 +14,9 @@ build:
       - asdf global uv latest
       - uv venv $READTHEDOCS_VIRTUALENV_PATH
       - UV_PROJECT_ENVIRONMENT=$READTHEDOCS_VIRTUALENV_PATH uv pip install -r doc/requirements.txt
+      # replace mujoco.mjx.third_party.mujoco_warp import paths with mujoco_warp
+      - |
+        find mjx/mujoco/mjx/third_party/mujoco_warp -type f -exec sed -i 's/mujoco\.mjx\.third_party\.mujoco_warp/mujoco_warp/g' {} \;
       - UV_PROJECT_ENVIRONMENT=$READTHEDOCS_VIRTUALENV_PATH uv pip install mjx/mujoco/mjx/third_party/mujoco_warp
     install:
       - "true"  # skip
diff --git a/doc/conf.py b/doc/conf.py
index b3f2f422..2cb148a9 100644
--- a/doc/conf.py
+++ b/doc/conf.py
@@ -24,7 +24,7 @@ import sys
 
 sys.path.insert(0, os.path.abspath('../'))
 sys.path.append(os.path.abspath('ext'))
-sys.path.insert(0, os.path.abspath('../mjx/mujoco/mjx/third_party/mujoco_warp'))
+sys.path.insert(0, os.path.abspath('../mjx/mujoco/mjx/third_party'))
 
 from sphinxcontrib import katex  # pylint: disable=g-import-not-at-top
 from sphinxcontrib import youtube  # pylint: disable=g-import-not-at-top,unused-import