diff --git a/packages/x-data-grid-pro/src/tests/rowEditing.DataGridPro.test.tsx b/packages/x-data-grid-pro/src/tests/rowEditing.DataGridPro.test.tsx
index d79e37a5f9807..450e62a54c633 100644
--- a/packages/x-data-grid-pro/src/tests/rowEditing.DataGridPro.test.tsx
+++ b/packages/x-data-grid-pro/src/tests/rowEditing.DataGridPro.test.tsx
@@ -1385,9 +1385,7 @@ describe('<DataGridPro /> - Row editing', () => {
       return (
         <Portal>
           <input
-            ref={(ref) => {
-              setInputRef(ref);
-            }}
+            ref={setInputRef}
             data-testid="input"
           />
         </Portal>