diff --git a/docker/Dockerfile.python3.12 b/docker/Dockerfile.python3.12
index 002e8ed3b..2c42853c5 100644
--- a/docker/Dockerfile.python3.12
+++ b/docker/Dockerfile.python3.12
@@ -58,14 +58,14 @@ RUN source $HOME/.local/bin/env && \
 # Install STA (Sliding Tile Attention)
 RUN source $HOME/.local/bin/env && \
     source /opt/venv/bin/activate && \
-    cd csrc/attn/ && \
+    cd csrc/attn && \
     git submodule update --init --recursive && \
     python setup_sta.py install
 
 # Install VSA 
 RUN source $HOME/.local/bin/env && \
     source /opt/venv/bin/activate && \
-    cd csrc/attn/ && \
+    cd csrc/attn && \
     git submodule update --init --recursive && \
     python setup_vsa.py install
 
diff --git a/pyproject.toml b/pyproject.toml
index 65ce83ac0..75506e14f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -49,6 +49,9 @@ dependencies = [
     "av",
 ]
 
+[tool.uv]
+extra-index-url = ["https://download.pytorch.org/whl/cu128"]
+
 [project.optional-dependencies]
 
 # flash-attn: pip install flash-attn==2.7.4.post1 --no-cache-dir --no-build-isolation