From 137738997d5c54a871db42a75fc4977146d1ab45 Mon Sep 17 00:00:00 2001
From: Jbleezy <thejbirdjustin@yahoo.com>
Date: Mon, 3 Jan 2022 05:23:42 -0800
Subject: [PATCH] Fix melee weapon not always switching back to correct weapon
 Optimize melee weapon switch code

---
 scripts/zm/_zm_reimagined.gsc | 19 ++++++-------------
 1 file changed, 6 insertions(+), 13 deletions(-)

diff --git a/scripts/zm/_zm_reimagined.gsc b/scripts/zm/_zm_reimagined.gsc
index c60ce163..57a6155c 100644
--- a/scripts/zm/_zm_reimagined.gsc
+++ b/scripts/zm/_zm_reimagined.gsc
@@ -1743,28 +1743,21 @@ melee_weapon_switch_watcher()
 		melee_wep = self get_player_melee_weapon();
 		curr_wep = self getCurrentWeapon();
 
+		if(curr_wep != "none" && !is_offhand_weapon(curr_wep))
+		{
+			prev_wep = curr_wep;
+		}
+
 		if(self actionSlotTwoButtonPressed() && !self hasWeapon("equip_dieseldrone_zm"))
 		{
 			if(curr_wep != melee_wep)
 			{
-				prev_wep = curr_wep;
 				self switchToWeapon(melee_wep);
 			}
 			else
 			{
-				if(isDefined(prev_wep) && self hasWeapon(prev_wep))
-				{
-					self switchToWeapon(prev_wep);
-				}
-				else
-				{
-					self switchToWeapon(self getWeaponsListPrimaries()[0]);
-				}
-
-				prev_wep = undefined;
+				self maps/mp/zombies/_zm_weapons::switch_back_primary_weapon(prev_wep);
 			}
-
-			self waittill("weapon_change");
 		}
 
 		wait 0.05;