chore(neoforge): Change to just keep track of the boolean

neoforged · Feb 2, 2025 · 2ea2e0c · 2ea2e0c
1 parent b47d554
commit 2ea2e0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/patches/net/minecraft/client/Minecraft.java.patch b/patches/net/minecraft/client/Minecraft.java.patch
@@ -333,7 +333,7 @@
          this.narrator.clear();
          this.clientLevelTeardownInProgress = true;
 
-+        var connection = this.getConnection() != null ? this.getConnection().getConnection() : null; // Neo: Hold connection to revert after shutdown
++        var shouldRevertRegistriesToFrozen = this.getConnection() != null && this.getConnection().getConnection() != null; // Neo: Track whether to revert registries after disconnect
          try {
              this.updateScreenAndTick(p_320248_);
              if (this.level != null) {
@@ -345,7 +345,7 @@
          }
 
          SkullBlockEntity.clear();
-+        if(connection != null) net.neoforged.neoforge.registries.RegistryManager.revertToFrozen(); // Neo: Revert registries to frozen on disconnect
++        if(shouldRevertRegistriesToFrozen) net.neoforged.neoforge.registries.RegistryManager.revertToFrozen(); // Neo: Revert registries to frozen on disconnect
      }
 
      public void clearDownloadedResourcePacks() {