diff --git a/dependencies.yaml b/dependencies.yaml
index 4210840d7..225635131 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -33,7 +33,7 @@ files:
   docs:
     output: none
     includes:
-      - cudatoolkit
+      - cuda_version
       - docs
       - py_version
       - pytorch_cpu