diff --git a/.gitlab/common/test_infra_version.yml b/.gitlab/common/test_infra_version.yml
index 31e364287c40c..7cbe57e233c71 100644
--- a/.gitlab/common/test_infra_version.yml
+++ b/.gitlab/common/test_infra_version.yml
@@ -4,4 +4,4 @@ variables:
   # and check the job creating the image to make sure you have the right SHA prefix
   TEST_INFRA_DEFINITIONS_BUILDIMAGES_SUFFIX: ""
   # Make sure to update test-infra-definitions version in go.mod as well
-  TEST_INFRA_DEFINITIONS_BUILDIMAGES: 08b9d8e048c1
+  TEST_INFRA_DEFINITIONS_BUILDIMAGES: 5898e20717d4
diff --git a/test/new-e2e/go.mod b/test/new-e2e/go.mod
index 496ee8b6854c3..9fc605ff12b1f 100644
--- a/test/new-e2e/go.mod
+++ b/test/new-e2e/go.mod
@@ -59,7 +59,7 @@ require (
 	// `TEST_INFRA_DEFINITIONS_BUILDIMAGES` matches the commit sha in the module version
 	// Example: 	github.com/DataDog/test-infra-definitions v0.0.0-YYYYMMDDHHmmSS-0123456789AB
 	// => TEST_INFRA_DEFINITIONS_BUILDIMAGES: 0123456789AB
-	github.com/DataDog/test-infra-definitions v0.0.0-20241007132100-08b9d8e048c1
+	github.com/DataDog/test-infra-definitions v0.0.0-20241010125334-5898e20717d4
 	github.com/aws/aws-sdk-go-v2 v1.32.0
 	github.com/aws/aws-sdk-go-v2/config v1.27.40
 	github.com/aws/aws-sdk-go-v2/service/ec2 v1.164.2
diff --git a/test/new-e2e/go.sum b/test/new-e2e/go.sum
index 4567e90b8819f..f6e761c6a1f7e 100644
--- a/test/new-e2e/go.sum
+++ b/test/new-e2e/go.sum
@@ -16,8 +16,8 @@ github.com/DataDog/datadog-go/v5 v5.5.0 h1:G5KHeB8pWBNXT4Jtw0zAkhdxEAWSpWH00geHI
 github.com/DataDog/datadog-go/v5 v5.5.0/go.mod h1:K9kcYBlxkcPP8tvvjZZKs/m1edNAUFzBbdpTUKfCsuw=
 github.com/DataDog/mmh3 v0.0.0-20200805151601-30884ca2197a h1:m9REhmyaWD5YJ0P53ygRHxKKo+KM+nw+zz0hEdKztMo=
 github.com/DataDog/mmh3 v0.0.0-20200805151601-30884ca2197a/go.mod h1:SvsjzyJlSg0rKsqYgdcFxeEVflx3ZNAyFfkUHP0TxXg=
-github.com/DataDog/test-infra-definitions v0.0.0-20241007132100-08b9d8e048c1 h1:vQFrWWJ8BUWqHpkUP4bu81ZVsk/2vPsykdiilL5LJCk=
-github.com/DataDog/test-infra-definitions v0.0.0-20241007132100-08b9d8e048c1/go.mod h1:spBV6LBmaH17vb/oe6n2kVfsuEUR+eQGsoaCGJclHF0=
+github.com/DataDog/test-infra-definitions v0.0.0-20241010125334-5898e20717d4 h1:6hWrBotT8nXQ385gH0HeV5GvtL8sgPzjfVNIaDha3io=
+github.com/DataDog/test-infra-definitions v0.0.0-20241010125334-5898e20717d4/go.mod h1:spBV6LBmaH17vb/oe6n2kVfsuEUR+eQGsoaCGJclHF0=
 github.com/DataDog/viper v1.13.5 h1:SZMcyMknYQN2jRY/40A16gUXexlNJOI8sDs1cWZnI64=
 github.com/DataDog/viper v1.13.5/go.mod h1:wDdUVJ2SHaMaPrCZrlRCObwkubsX8j5sme3LaR/SGTc=
 github.com/DataDog/zstd v1.5.5 h1:oWf5W7GtOLgp6bciQYDmhHHjdhYkALu6S/5Ni9ZgSvQ=