diff --git a/apps/bot-discord/src/utils/trainAISelectMenu.js b/apps/bot-discord/src/utils/trainAISelectMenu.js
index 7406f98..a0c46d1 100644
--- a/apps/bot-discord/src/utils/trainAISelectMenu.js
+++ b/apps/bot-discord/src/utils/trainAISelectMenu.js
@@ -58,7 +58,7 @@ export default async function trainAISelectMenu(
 	const voteId = interaction.targetMessage ? interaction.targetMessage.id :
 		interaction.message.reference.messageId;
 	collector.on('collect', (i) => {
-		i.message.edit({ content: 'Sent training data to server.', components: [] });
+		interaction.editReply({ content: 'Sent training data to server.', components: [] });
 
 		const existingVote = interaction.client.trainingVotes.get(voteId);
 		if (existingVote) clearTimeout(existingVote);