diff --git a/x-pack/platform/packages/private/ml/url_state/src/url_state.tsx b/x-pack/platform/packages/private/ml/url_state/src/url_state.tsx
index e24176321094a..9727c63b0394e 100644
--- a/x-pack/platform/packages/private/ml/url_state/src/url_state.tsx
+++ b/x-pack/platform/packages/private/ml/url_state/src/url_state.tsx
@@ -291,7 +291,7 @@ export const useUrlStateService = <K extends Accessor, T>(
     ? state?.[optionsRef.current.pageKey]
     : state;
 
-  const setCallback = useRef<typeof setState>();
+  const setCallback = useRef<typeof setState>(setState);
 
   useEffect(() => {
     setCallback.current = setState;