diff --git a/ai_commands.py b/ai_commands.py
index 13a845e..a725b11 100644
--- a/ai_commands.py
+++ b/ai_commands.py
@@ -88,7 +88,7 @@ class Events(commands.Cog):
                     self.logger.info(ctx.message.author.id)
                     self.logger.info(superfryta)
                     if ctx.message.author.id == superfryta[0]:
-                        self.armia[superfryta[2]] = dm_mode_arg
+                        self.armia[ctx.message.author.id] = dm_mode_arg
                         self.logger.info(self.armia)
                         await ctx.reply("Weszlo")
                         return