From c9c96006cb0791da0bf62e0d9496c32f8516a189 Mon Sep 17 00:00:00 2001
From: amevarashi <kp-karasu@yandex.ru>
Date: Sat, 4 Jun 2022 13:40:40 +0500
Subject: [PATCH] Check comp before copying enumerable

---
 RJWSexperience/RJWSexperience/Patches/GetGizmos.cs | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/RJWSexperience/RJWSexperience/Patches/GetGizmos.cs b/RJWSexperience/RJWSexperience/Patches/GetGizmos.cs
index 6c975ac..5cdf052 100644
--- a/RJWSexperience/RJWSexperience/Patches/GetGizmos.cs
+++ b/RJWSexperience/RJWSexperience/Patches/GetGizmos.cs
@@ -12,17 +12,19 @@ namespace RJWSexperience
 			if (Find.Selector.NumSelected > 1)
 				return;
 
-			__result = AddHistoryGizmo(__instance, __result);
+			SexHistoryComp history = __instance.TryGetComp<SexHistoryComp>();
+			if (history == null)
+				return;
+
+			__result = AddHistoryGizmo(history, __result);
 		}
 
-		private static IEnumerable<Gizmo> AddHistoryGizmo(Pawn pawn, IEnumerable<Gizmo> gizmos)
+		private static IEnumerable<Gizmo> AddHistoryGizmo(SexHistoryComp history, IEnumerable<Gizmo> gizmos)
 		{
 			foreach (Gizmo gizmo in gizmos)
 				yield return gizmo;
 
-			SexHistoryComp history = pawn.TryGetComp<SexHistoryComp>();
-			if (history != null)
-				yield return history.Gizmo;
+			yield return history.Gizmo;
 		}
 	}
 }