diff --git a/MiraAPI.Example/Buttons/MeetingButton.cs b/MiraAPI.Example/Buttons/MeetingButton.cs index 4b4f087..a5de0e9 100644 --- a/MiraAPI.Example/Buttons/MeetingButton.cs +++ b/MiraAPI.Example/Buttons/MeetingButton.cs @@ -21,7 +21,7 @@ public class MeetingButton : CustomActionButton public override bool Enabled(RoleBehaviour? role) { - return PlayerControl.LocalPlayer.HasModifier(); + return PlayerControl.LocalPlayer != null && PlayerControl.LocalPlayer.HasModifier(); } protected override void OnClick() diff --git a/MiraAPI/Utilities/Extensions.cs b/MiraAPI/Utilities/Extensions.cs index 156c87f..38f2567 100644 --- a/MiraAPI/Utilities/Extensions.cs +++ b/MiraAPI/Utilities/Extensions.cs @@ -133,11 +133,12 @@ public static bool IsCustom(this OptionBehaviour optionBehaviour) } component = player.GetComponent(); - if (component) + if (component == null) { - ModifierComponents[player] = component; + return null; } + ModifierComponents[player] = component; return component; }