From d7631c51b686bb0545476222905b3b1271f09879 Mon Sep 17 00:00:00 2001
From: rogeraabbccdd <rogeraabbccdd@gmail.com>
Date: Wed, 3 Jul 2024 00:51:35 +0800
Subject: [PATCH] Fixed agent menu.

---
 Commands.cs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Commands.cs b/Commands.cs
index 5935462d..1f3c4c78 100644
--- a/Commands.cs
+++ b/Commands.cs
@@ -474,7 +474,7 @@ namespace WeaponPaints
 				var selectedPaintName = option.Text;
 				var selectedAgent = agentsList.FirstOrDefault(g =>
 					g.ContainsKey("agent_name") &&
-					g["agent_name"] != null && g["agent_name"]!.ToString() == selectedPaintName &&
+					g["agent_name"] != null && g["agent_name"]!.ToString().Contains(selectedPaintName) == true &&
 					g["team"] != null && (int)(g["team"]!) == player.TeamNum);
 
 				if (selectedAgent == null) return;
@@ -502,7 +502,7 @@ namespace WeaponPaints
 
 					if (!string.IsNullOrEmpty(Localizer["wp_agent_menu_select"]))
 					{
-						player!.Print(Localizer["wp_agent_menu_select", selectedPaintName]);
+						player!.Print(Localizer["wp_agent_menu_select", selectedAgent?["agent_name"] ?? selectedPaintName]);
 					}
 
 					if (player.TeamNum == 3)
@@ -524,6 +524,7 @@ namespace WeaponPaints
 						_ = Task.Run(async () =>
 						{
 							await weaponSync.SyncAgentToDatabase(playerInfo);
+							GivePlayerAgent(player);
 						});
 					}
 				};