diff --git a/models/document/src/migration.ts b/models/document/src/migration.ts
index dffaeb7a44..647d7ed4ba 100644
--- a/models/document/src/migration.ts
+++ b/models/document/src/migration.ts
@@ -355,6 +355,10 @@ export const documentOperation: MigrateOperation = {
       {
         state: 'migrateEmbeddings',
         func: migrateEmbeddings
+      },
+      {
+        state: 'migrateEmbeddingsRefs',
+        func: migrateEmbeddingsRefs
       }
     ])
   },
@@ -370,3 +374,13 @@ async function migrateEmbeddings (client: MigrationClient): Promise<void> {
   )
   await client.move(DOMAIN_DOCUMENT, { _class: attachment.class.Embedding }, DOMAIN_ATTACHMENT)
 }
+
+async function migrateEmbeddingsRefs (client: MigrationClient): Promise<void> {
+  const _class = 'document:class:DocumentEmbedding'
+
+  await client.update(DOMAIN_ACTIVITY, { attachedToClass: _class }, { attachedToClass: attachment.class.Embedding })
+  await client.update(DOMAIN_ACTIVITY, { objectClass: _class }, { objectClass: attachment.class.Embedding })
+  await client.update(DOMAIN_NOTIFICATION, { attachedToClass: _class }, { attachedToClass: attachment.class.Embedding })
+  await client.update(DOMAIN_TX, { objectClass: _class }, { objectClass: attachment.class.Embedding })
+  await client.update(DOMAIN_TX, { 'tx.objectClass': _class }, { 'tx.objectClass': attachment.class.Embedding })
+}