diff --git a/x-pack/test_serverless/api_integration/test_suites/common/index_management/datastreams.ts b/x-pack/test_serverless/api_integration/test_suites/common/index_management/datastreams.ts
index 12151f1b169d..099e1b9b3c1a 100644
--- a/x-pack/test_serverless/api_integration/test_suites/common/index_management/datastreams.ts
+++ b/x-pack/test_serverless/api_integration/test_suites/common/index_management/datastreams.ts
@@ -135,11 +135,12 @@ export default function ({ getService }: FtrProviderContext) {
 
       it('updates the data retention of a DS', async () => {
         const { body, status } = await supertestWithoutAuth
-          .put(`${API_BASE_PATH}/data_streams/${testDataStreamName}/data_retention`)
+          .put(`${API_BASE_PATH}/data_streams/data_retention`)
           .set(internalReqHeader)
           .set(roleAuthc.apiKeyHeader)
           .send({
             dataRetention: '7d',
+            dataStreams: [testDataStreamName],
           });
         svlCommonApi.assertResponseStatusCode(200, status, body);
 
@@ -148,10 +149,12 @@ export default function ({ getService }: FtrProviderContext) {
 
       it('sets data retention to infinite', async () => {
         const { body, status } = await supertestWithoutAuth
-          .put(`${API_BASE_PATH}/data_streams/${testDataStreamName}/data_retention`)
+          .put(`${API_BASE_PATH}/data_streams/data_retention`)
           .set(internalReqHeader)
           .set(roleAuthc.apiKeyHeader)
-          .send({});
+          .send({
+            dataStreams: [testDataStreamName],
+          });
         svlCommonApi.assertResponseStatusCode(200, status, body);
 
         // Providing an infinite retention might not be allowed for a given project,