From a7976c9fc9ee2aad86ffa75b5192b84870e8201b Mon Sep 17 00:00:00 2001
From: kosmik641 <minecraftgame2002@gmail.com>
Date: Fri, 15 Nov 2024 17:00:20 +0300
Subject: [PATCH] =?UTF-8?q?=D0=A3=D0=B4=D0=B0=D0=BB=D0=B5=D0=BD=D0=B8?=
 =?UTF-8?q?=D0=B5=20=D1=83=D0=B4=D0=BE=D1=87=D0=BA=D0=B8=20=D0=B8=D0=B7=20?=
 =?UTF-8?q?=D1=82=D0=B5=D0=BB=D0=B5=D0=B6=D0=BA=D0=B8=20=D1=81=D1=80=D0=B0?=
 =?UTF-8?q?=D0=B7=D1=83=20=D0=BF=D0=BE=20=D0=B5=D1=91=20=D0=BE=D1=82=D0=BA?=
 =?UTF-8?q?=D0=BB=D1=8E=D1=87=D0=B5=D0=BD=D0=B8=D1=8E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lua/entities/gmod_track_udochka/init.lua | 4 +++-
 lua/entities/gmod_train_bogey/init.lua   | 6 +-----
 2 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/lua/entities/gmod_track_udochka/init.lua b/lua/entities/gmod_track_udochka/init.lua
index 3186b87..c6f9e1c 100644
--- a/lua/entities/gmod_track_udochka/init.lua
+++ b/lua/entities/gmod_track_udochka/init.lua
@@ -22,13 +22,15 @@ local function removeWeldBetweenEnts(ent1,ent2)
 	end
 end
 function ENT:Use(ply)
-	if self.Coupled then
+	if IsValid(self.Coupled) then
 		sound.Play("buttons/lever8.wav",self:GetPos())
 		removeWeldBetweenEnts(self,self.Coupled)
 		removeWeldBetweenEnts(self.Coupled,self)
 
+		self.Coupled.Connectors[self.ConnectorID] = nil
 		self.Timer = CurTime()+2
 	end
+	self.ConnectorID = nil
 	self.Coupled = nil
 	if ( self:IsPlayerHolding() ) then return end
 	if ply.PickupObject then ply:PickupObject( self ) end
diff --git a/lua/entities/gmod_train_bogey/init.lua b/lua/entities/gmod_train_bogey/init.lua
index ac6ab1a..de2cc44 100644
--- a/lua/entities/gmod_train_bogey/init.lua
+++ b/lua/entities/gmod_train_bogey/init.lua
@@ -392,17 +392,13 @@ function ENT:CheckContact(pos,dir,id,cpos)
                 traceEnt:SetPos(self:LocalToWorld(cpos))
                 traceEnt:SetAngles(self:GetAngles())
                 traceEnt.Coupled = self
+                traceEnt.ConnectorID = id
                 sound.Play("udochka_connect.wav",traceEnt:GetPos())
                 self.Connectors[id] = traceEnt
                 DropEntityIfHeld(traceEnt)
             end
         end
         return false
-    elseif self.Connectors[id] == traceEnt then
-        if traceEnt.Coupled == nil then
-            self.Connectors[id] = nil
-        end
-        return false
     elseif traceEnt:GetClass() == "player" and self.Voltage > 40 then
         local pPos = traceEnt:GetPos()
         self.VoltageDropByTouch = (self.VoltageDropByTouch or 0) + 1