diff --git a/env/hpc/requirements.txt b/env/hpc/requirements.txt
index 819ea32..84515f9 100644
--- a/env/hpc/requirements.txt
+++ b/env/hpc/requirements.txt
@@ -14,6 +14,7 @@ mediapy==1.2.6
 optax>=0.2.6
 pyopengl>=3.1.10
 pyopengl-accelerate>=3.1.10
+pyyaml>=6.0
 tyro>=1.0.10
 wandb==0.24.2
 torch>=2.4.0
diff --git a/scripts/hpc/export_requirements.py b/scripts/hpc/export_requirements.py
index 5908ad3..62c1298 100644
--- a/scripts/hpc/export_requirements.py
+++ b/scripts/hpc/export_requirements.py
@@ -13,7 +13,7 @@ import re
 import sys
 from pathlib import Path
 
-ROOT = Path(__file__).parent.parent
+ROOT = Path(__file__).resolve().parents[2]
 
 
 def normalise(name: str) -> str: