diff --git a/packages/presentation/src/overlays/schema/PostMessageSchema.tsx b/packages/presentation/src/overlays/schema/PostMessageSchema.tsx
index 819b51a06..c269cb04c 100644
--- a/packages/presentation/src/overlays/schema/PostMessageSchema.tsx
+++ b/packages/presentation/src/overlays/schema/PostMessageSchema.tsx
@@ -1,10 +1,9 @@
 import type {UnresolvedPath} from '@repo/visual-editing-helpers'
 import type {ClientPerspective} from '@sanity/client'
-import {getPublishedId} from '@sanity/client/csm'
 import {useRootTheme} from '@sanity/ui'
 import {memo, useEffect} from 'react'
 import {API_VERSION} from '../../constants'
-import {useClient, useWorkspace} from '../../internals'
+import {getPublishedId, useClient, useWorkspace} from '../../internals'
 import {getBundlePerspective} from '../../loader/utils'
 import type {VisualEditingConnection} from '../../types'
 import {extractSchema} from './extract'