diff --git a/packages/x-data-grid-pro/src/tests/dataSource.DataGridPro.test.tsx b/packages/x-data-grid-pro/src/tests/dataSource.DataGridPro.test.tsx
index a3ed9c493c893..5e44522400831 100644
--- a/packages/x-data-grid-pro/src/tests/dataSource.DataGridPro.test.tsx
+++ b/packages/x-data-grid-pro/src/tests/dataSource.DataGridPro.test.tsx
@@ -106,38 +106,32 @@ describeSkipIf(isJSDOM)('<DataGridPro /> - Data source', () => {
   });
 
   it('should re-fetch the data on sort change', async () => {
-    const { setProps } = render(<TestDataSource />);
-    await waitFor(() => {
-      expect(fetchRowsSpy.callCount).to.equal(1);
-    });
-    setProps({ sortModel: [{ field: 'name', sort: 'asc' }] });
+    render(<TestDataSource />);
+    expect(fetchRowsSpy.callCount).to.equal(1);
+    apiRef.current.setSortModel([{ field: 'name', sort: 'asc' }]);
     await waitFor(() => {
       expect(fetchRowsSpy.callCount).to.equal(2);
     });
   });
 
   it('should re-fetch the data on pagination change', async () => {
-    const { setProps } = render(<TestDataSource />);
-    await waitFor(() => {
-      expect(fetchRowsSpy.callCount).to.equal(1);
-    });
-    setProps({ paginationModel: { page: 1, pageSize: 10 } });
+    render(<TestDataSource />);
+    expect(fetchRowsSpy.callCount).to.equal(1);
+    apiRef.current.setPaginationModel({ page: 1, pageSize: 10 });
     await waitFor(() => {
       expect(fetchRowsSpy.callCount).to.equal(2);
     });
   });
 
   it('should re-fetch the data once if multiple models have changed', async () => {
-    const { setProps } = render(<TestDataSource />);
-    await waitFor(() => {
-      expect(fetchRowsSpy.callCount).to.equal(1);
-    });
+    render(<TestDataSource />);
+    expect(fetchRowsSpy.callCount).to.equal(1);
 
-    setProps({
-      paginationModel: { page: 1, pageSize: 10 },
-      sortModel: [{ field: 'name', sort: 'asc' }],
-      filterModel: { items: [{ field: 'name', value: 'John', operator: 'contains' }] },
+    apiRef.current.setFilterModel({
+      items: [{ field: 'name', value: 'John', operator: 'contains' }],
     });
+    apiRef.current.setSortModel([{ field: 'name', sort: 'asc' }]);
+    apiRef.current.setPaginationModel({ page: 1, pageSize: 10 });
 
     await waitFor(() => {
       expect(fetchRowsSpy.callCount).to.equal(2);
@@ -217,7 +211,9 @@ describeSkipIf(isJSDOM)('<DataGridPro /> - Data source', () => {
       await waitFor(() => {
         expect(fetchRowsSpy.callCount).to.equal(2);
       });
-      expect(cache.size).to.equal(2);
+      await waitFor(() => {
+        expect(cache.size).to.equal(2);
+      });
 
       const dataRow2 = await screen.findByText(
         (_, el) => el?.getAttribute('data-rowindex') === '0' && el !== dataRow1,
diff --git a/packages/x-data-grid-pro/src/tests/dataSourceLazyLoader.DataGridPro.test.tsx b/packages/x-data-grid-pro/src/tests/dataSourceLazyLoader.DataGridPro.test.tsx
index 0e8daf15e654a..ae7568a1c8e97 100644
--- a/packages/x-data-grid-pro/src/tests/dataSourceLazyLoader.DataGridPro.test.tsx
+++ b/packages/x-data-grid-pro/src/tests/dataSourceLazyLoader.DataGridPro.test.tsx
@@ -86,15 +86,15 @@ describeSkipIf(isJSDOM)('<DataGridPro /> - Data source lazy loader', () => {
   });
 
   it('should re-fetch the data once if multiple models have changed', async () => {
-    const { setProps } = render(<TestDataSourceLazyLoader />);
+    render(<TestDataSourceLazyLoader />);
     await waitFor(() => {
       expect(fetchRowsSpy.callCount).to.equal(1);
     });
 
-    setProps({
-      sortModel: [{ field: 'name', sort: 'asc' }],
-      filterModel: { items: [{ field: 'name', value: 'John', operator: 'contains' }] },
+    apiRef.current.setFilterModel({
+      items: [{ field: 'name', value: 'John', operator: 'contains' }],
     });
+    apiRef.current.setSortModel([{ field: 'name', sort: 'asc' }]);
 
     await waitFor(() => {
       expect(fetchRowsSpy.callCount).to.equal(2);
diff --git a/packages/x-data-grid-pro/src/tests/dataSourceTreeData.DataGridPro.test.tsx b/packages/x-data-grid-pro/src/tests/dataSourceTreeData.DataGridPro.test.tsx
index b095b73776dee..cdfd498bc26d8 100644
--- a/packages/x-data-grid-pro/src/tests/dataSourceTreeData.DataGridPro.test.tsx
+++ b/packages/x-data-grid-pro/src/tests/dataSourceTreeData.DataGridPro.test.tsx
@@ -114,11 +114,11 @@ describeSkipIf(isJSDOM)('<DataGridPro /> - Data source tree data', () => {
   });
 
   it('should re-fetch the data on pagination change', async () => {
-    const { setProps } = render(<TestDataSource />);
+    render(<TestDataSource />);
     await waitFor(() => {
       expect(fetchRowsSpy.callCount).to.equal(1);
     });
-    setProps({ paginationModel: { page: 1, pageSize: 10 } });
+    apiRef.current.setPaginationModel({ page: 1, pageSize: 10 });
     await waitFor(() => {
       expect(fetchRowsSpy.callCount).to.equal(2);
     });