Merge pull request #31 from Cody-Spring/master

Multiplayer Patch Update
rheirman · May 4, 2020 · ded78cd · ded78cd
2 parents ecb29e0 + a951384
commit ded78cd
Showing 1 changed file with 9 additions and 12 deletions.
diff --git a/v1.1/Source/Giddy-up-Core/ModExtensions/MultiplayerPatch.cs b/v1.1/Source/Giddy-up-Core/ModExtensions/MultiplayerPatch.cs
@@ -1,4 +1,4 @@
-using System;
+using System;
 using System.Collections.Generic;
 using System.Reflection;
 using System.Reflection.Emit;
@@ -49,26 +49,23 @@ static void ExtendedDataStoragePostfix()
 
         static void GetExtendedDataForPostfix(Pawn pawn, object __result)
         {
-            if (forward.ContainsKey(pawn))
+            if (!forward.ContainsKey(pawn))
             {
-                return;
-            }
-            forward.Add(pawn, __result);
-            if (backward.ContainsKey(pawn))
+                forward.Add(pawn, __result);
+            }    
+            if (!backward.ContainsKey(__result))
             {
-                return;
+                backward.Add(__result, pawn);
             }
-            backward.Add(__result, pawn);
         }
 
         static void DeleteExtendedDataForPostfix(Pawn pawn)
         {
-            if (!forward.ContainsKey(pawn))
+            if (forward.ContainsKey(pawn))
             {
-                return;
+                backward.Remove(forward[pawn]);
+                forward.Remove(pawn);
             }
-            backward.Remove(forward[pawn]);
-            forward.Remove(pawn);
         }
 
         static void ExtendedPawnData(SyncWorker sync, ref object obj)