Set current command to the last used

2 files changed, 5 insertions, 2 deletions

diff --git a/lua/aotrix/build.lua b/lua/aotrix/build.lua
index 96a7f75..8408814 100644
--- a/lua/aotrix/build.lua
+++ b/lua/aotrix/build.lua
@@ -20,7 +20,10 @@ local COMMANDS = {}
 local personalCommands = require("aotrix.personal_commands")
 
 COMMANDS.AddTempCommand = {
-    command = function() return personalCommands.createCommand(COMMANDS) end,
+    command = function()
+        CURRENT_COMMAND = personalCommands.createCommand(COMMANDS).command
+        return CURRENT_COMMAND()
+    end,
     description = function() return "Add temporary command" end
 }
 
diff --git a/lua/aotrix/personal_commands.lua b/lua/aotrix/personal_commands.lua
index e1094d5..16364d2 100644
--- a/lua/aotrix/personal_commands.lua
+++ b/lua/aotrix/personal_commands.lua
@@ -6,6 +6,6 @@ return {
             command = function() return selectedCommand end,
             description = function() return selectedDescription end
         }
-        return COMMANDS.Flower.command()
+        return COMMANDS.Flower
     end
 }