Fixed Trails Tool Undo

Facepunch · Jan 19, 2014 · 9aebac4 · 9aebac4
1 parent 2a2f269
commit 9aebac4
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/garrysmod/gamemodes/sandbox/entities/weapons/gmod_tool/stools/trails.lua b/garrysmod/gamemodes/sandbox/entities/weapons/gmod_tool/stools/trails.lua
@@ -17,7 +17,7 @@ cleanup.Register( "trails" )
 
 local function SetTrails( Player, Entity, Data )
 
-	if ( Entity.SToolTrail ) then
+	if ( IsValid( Entity.SToolTrail ) ) then
 
 		Entity.SToolTrail:Remove()
 		Entity.SToolTrail = nil
@@ -96,14 +96,17 @@ function TOOL:LeftClick( trace )
 
 	end
 
-	local Trail = SetTrails( self:GetOwner(), trace.Entity, { Color = Color( r, g, b, a ), 
-																	Length = length, 
-																	StartSize = startsize, 
-																	EndSize = endsize,
-																	Material = Mat } )
+	local Trail = SetTrails( self:GetOwner(), trace.Entity, {
+		Color = Color( r, g, b, a ), 
+		Length = length, 
+		StartSize = startsize, 
+		EndSize = endsize,
+		Material = Mat
+	} )
+
 	undo.Create("Trail")
 		undo.AddEntity( Trail )
-		undo.SetPlayer( ply )
+		undo.SetPlayer( self:GetOwner() )
 	undo.Finish()
 
 	return true