diff --git a/packages/x-data-grid-premium/src/hooks/utils/useGridApiContext.ts b/packages/x-data-grid-premium/src/hooks/utils/useGridApiContext.ts
index de89813383fde..a998dced7aa03 100644
--- a/packages/x-data-grid-premium/src/hooks/utils/useGridApiContext.ts
+++ b/packages/x-data-grid-premium/src/hooks/utils/useGridApiContext.ts
@@ -1,4 +1,6 @@
-import { useGridApiContext as useCommunityGridApiContext } from '@mui/x-data-grid';
+import { RefObject } from '@mui/x-internals/types';
+import { GridApiCommon, useGridApiContext as useCommunityGridApiContext } from '@mui/x-data-grid';
 import { GridApiPremium } from '../../models/gridApiPremium';
 
-export const useGridApiContext = useCommunityGridApiContext<GridApiPremium>;
+export const useGridApiContext: <Api extends GridApiCommon = GridApiPremium>() => RefObject<Api> =
+  useCommunityGridApiContext;
diff --git a/packages/x-data-grid-pro/src/hooks/utils/useGridApiContext.ts b/packages/x-data-grid-pro/src/hooks/utils/useGridApiContext.ts
index bd5f4f2a1b599..d3ee820aac6d4 100644
--- a/packages/x-data-grid-pro/src/hooks/utils/useGridApiContext.ts
+++ b/packages/x-data-grid-pro/src/hooks/utils/useGridApiContext.ts
@@ -1,4 +1,6 @@
-import { useGridApiContext as useCommunityGridApiContext } from '@mui/x-data-grid';
+import { RefObject } from '@mui/x-internals/types';
+import { GridApiCommon, useGridApiContext as useCommunityGridApiContext } from '@mui/x-data-grid';
 import { GridApiPro } from '../../models/gridApiPro';
 
-export const useGridApiContext = useCommunityGridApiContext<GridApiPro>;
+export const useGridApiContext: <Api extends GridApiCommon = GridApiPro>() => RefObject<Api> =
+  useCommunityGridApiContext;