diff --git a/apps/sim/app/(landing)/integrations/(shell)/[slug]/page.tsx b/apps/sim/app/(landing)/integrations/(shell)/[slug]/page.tsx
index 0b7c616dd80..e0ea9f9579b 100644
--- a/apps/sim/app/(landing)/integrations/(shell)/[slug]/page.tsx
+++ b/apps/sim/app/(landing)/integrations/(shell)/[slug]/page.tsx
@@ -17,7 +17,7 @@ import { IntegrationCtaButton } from '@/app/(landing)/integrations/(shell)/[slug
import { IntegrationFAQ } from '@/app/(landing)/integrations/(shell)/[slug]/components/integration-faq'
import { TemplateCardButton } from '@/app/(landing)/integrations/(shell)/[slug]/components/template-card-button'
import { IntegrationIcon } from '@/app/(landing)/integrations/components/integration-icon'
-import { getTemplatesForBlock } from '@/blocks/registry'
+import { getTemplatesForBlock } from '@/blocks/manifest'
const allIntegrations = INTEGRATIONS
const INTEGRATION_COUNT = allIntegrations.length
diff --git a/apps/sim/app/workspace/[workspaceId]/home/components/chat-context-kind-registry/chat-context-kind-registry.tsx b/apps/sim/app/workspace/[workspaceId]/home/components/chat-context-kind-registry/chat-context-kind-registry.tsx
index e01cbc5f80d..94c0dbdfe25 100644
--- a/apps/sim/app/workspace/[workspaceId]/home/components/chat-context-kind-registry/chat-context-kind-registry.tsx
+++ b/apps/sim/app/workspace/[workspaceId]/home/components/chat-context-kind-registry/chat-context-kind-registry.tsx
@@ -12,7 +12,7 @@ import { AgentSkillsIcon } from '@/components/icons'
import { getDocumentIcon } from '@/components/icons/document-icons'
import type { ChatContextKind, ChatMessageContext } from '@/app/workspace/[workspaceId]/home/types'
import { getBareIconStyle } from '@/blocks/icon-color'
-import { registry as blockRegistry } from '@/blocks/registry'
+import { getBlockDisplay } from '@/blocks/manifest'
interface RenderIconArgs {
context: ChatMessageContext
@@ -39,7 +39,7 @@ function renderWorkflowIcon({ className }: RenderIconArgs): ReactNode | null {
function renderIntegrationTile({ context, className }: RenderIconArgs): ReactNode | null {
if (context.kind !== 'integration') return null
if (!context.blockType) return null
- const block = blockRegistry[context.blockType]
+ const block = getBlockDisplay(context.blockType)
if (!block) return null
const Icon = block.icon
return