Have GetMenstruationComp(Hediff) use AllVaginas

2022-09-03 18:30:29 -07:00 · 2022-09-03 18:30:29 -07:00 · ffee2e8149
parent 49f52891fe
commit ffee2e8149
1 changed files with 1 additions and 1 deletions
--- a/1.3/source/RJW_Menstruation/RJW_Menstruation/HediffComps/MenstruationUtility.cs
+++ b/1.3/source/RJW_Menstruation/RJW_Menstruation/HediffComps/MenstruationUtility.cs
@ -46,7 +46,7 @@ namespace RJW_Menstruation

        public static HediffComp_Menstruation GetMenstruationComp(this Hediff vagina)
        {
-            if (vagina is Hediff_PartBaseNatural || vagina is Hediff_PartBaseArtifical)
+            if (VariousDefOf.AllVaginas.Contains(vagina?.def))
            {
                return vagina.TryGetComp<HediffComp_Menstruation>();
            }