diff --git a/pyproject.toml b/pyproject.toml
index 4bb65005..e74b8065 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -55,8 +55,8 @@ dependencies = [
     "rerun-sdk>=0.21.0",
     "termcolor>=2.4.0",
     "torch>=2.2.1",
+    "torchcodec>=0.2.1",
     "torchvision>=0.21.0",
-    "torchcodec>=0.1.0",
     "wandb>=0.16.3",
     "zarr>=2.17.0",
 ]