diff --git a/src/LegacySocket/messages.ts b/src/LegacySocket/messages.ts
index c7a972a..830ffb2 100644
--- a/src/LegacySocket/messages.ts
+++ b/src/LegacySocket/messages.ts
@@ -98,6 +98,8 @@ const makeMessagesSocket = (config: LegacySocketConfig) => {
 		Object.assign(content, attrs) // update message
 
 		ev.emit('messages.upsert', { messages: [message], type: 'replace' })
+
+		return message
 	}
 
 	const onMessage = (message: WAMessage, type: MessageUpdateType) => {