From 528302e07448396d0388eac90c4d6e0961bbe800 Mon Sep 17 00:00:00 2001
From: Michael Hillcox <michaelhillcox@gmail.com>
Date: Thu, 9 May 2024 15:02:49 +0100
Subject: [PATCH] fix: only change weather on the overworld

---
 .../java/com/sunekaer/toolkit/commands/TKCommand.java  | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/common/src/main/java/com/sunekaer/toolkit/commands/TKCommand.java b/common/src/main/java/com/sunekaer/toolkit/commands/TKCommand.java
index 2f5cece..1a484eb 100644
--- a/common/src/main/java/com/sunekaer/toolkit/commands/TKCommand.java
+++ b/common/src/main/java/com/sunekaer/toolkit/commands/TKCommand.java
@@ -23,14 +23,12 @@ public static void register(CommandDispatcher<CommandSourceStack> dispatcher, Co
         dispatcher.register(Commands.literal("toggledownfall")
                 .requires(cs -> cs.hasPermission(Commands.LEVEL_GAMEMASTERS))
                 .executes(c -> {
-                    ServerLevel level = c.getSource()
-                            .getPlayerOrException()
-                            .serverLevel();
+                    ServerLevel overworld = c.getSource().getServer().overworld();
 
-                    if (level.isRaining() || level.isThundering()) {
-                        level.setWeatherParameters(6000, 0, false, false);
+                    if (overworld.isRaining() || overworld.isThundering()) {
+                        overworld.setWeatherParameters(6000, 0, false, false);
                     } else {
-                        level.setWeatherParameters(0, 6000, true, false);
+                        overworld.setWeatherParameters(0, 6000, true, false);
                     }
 
                     return 1;