diff --git a/tests/e2e/infra/aws.yaml b/tests/e2e/infra/aws.yaml
index d73bf7f5..b512fc78 100644
--- a/tests/e2e/infra/aws.yaml
+++ b/tests/e2e/infra/aws.yaml
@@ -28,3 +28,4 @@ spec:
     install: false
   nvidiaDriver:
     install: true
+    branch: 550