From 972045c79332a4e14579e86f5bc31cece72bc810 Mon Sep 17 00:00:00 2001
From: Tyfon <29051038+tyfon7@users.noreply.github.com>
Date: Fri, 19 Apr 2024 16:15:15 -0700
Subject: [PATCH] Nit

---
 Patches/EditBuildScreenPatch.cs | 6 ++++++
 Plugin.cs                       | 3 +--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/Patches/EditBuildScreenPatch.cs b/Patches/EditBuildScreenPatch.cs
index f6ff868..edaeaac 100644
--- a/Patches/EditBuildScreenPatch.cs
+++ b/Patches/EditBuildScreenPatch.cs
@@ -12,6 +12,12 @@ namespace UIFixes
     {
         public static bool MoveForward;
 
+        public static void Enable()
+        {
+            new CloseScreenInterruptionPatch().Enable();
+            new ConfirmDiscardPatch().Enable();
+        }
+
         public class CloseScreenInterruptionPatch : ModulePatch
         {
             protected override MethodBase GetTargetMethod()
diff --git a/Plugin.cs b/Plugin.cs
index e449943..41a5abd 100644
--- a/Plugin.cs
+++ b/Plugin.cs
@@ -9,8 +9,7 @@ namespace UIFixes
         {
             Settings.Init(Config);
 
-            new EditBuildScreenPatch.CloseScreenInterruptionPatch().Enable();
-            new EditBuildScreenPatch.ConfirmDiscardPatch().Enable();
+            EditBuildScreenPatch.Enable();
             new TransferConfirmPatch().Enable();
             new MailReceiveAllPatch().Enable();
             ScrollPatches.Enable();