diff --git a/src/main/java/dev/micle/totemofreviving/ISideProxy.java b/src/main/java/dev/micle/totemofreviving/ISideProxy.java
index 9cdc96b..71a2e80 100644
--- a/src/main/java/dev/micle/totemofreviving/ISideProxy.java
+++ b/src/main/java/dev/micle/totemofreviving/ISideProxy.java
@@ -1,13 +1,13 @@
 package dev.micle.totemofreviving;
 
-import net.minecraft.client.multiplayer.ClientLevel;
 import net.minecraft.server.MinecraftServer;
 import net.minecraft.world.entity.player.Player;
+import net.minecraft.world.level.Level;
 
 public interface ISideProxy {
 	MinecraftServer getServer();
 	
 	Player getClientPlayer();
 	
-	ClientLevel getClientWorld();
+	Level getClientWorld();
 }
diff --git a/src/main/java/dev/micle/totemofreviving/SideProxy.java b/src/main/java/dev/micle/totemofreviving/SideProxy.java
index 511f93f..313c6d8 100644
--- a/src/main/java/dev/micle/totemofreviving/SideProxy.java
+++ b/src/main/java/dev/micle/totemofreviving/SideProxy.java
@@ -5,9 +5,11 @@ import dev.micle.totemofreviving.data.DataGenerators;
 import dev.micle.totemofreviving.network.Network;
 import dev.micle.totemofreviving.setup.Registration;
 import net.minecraft.client.Minecraft;
-import net.minecraft.client.multiplayer.ClientLevel;
 import net.minecraft.server.MinecraftServer;
 import net.minecraft.world.entity.player.Player;
+import net.minecraft.world.level.Level;
+import net.minecraftforge.api.distmarker.Dist;
+import net.minecraftforge.api.distmarker.OnlyIn;
 import net.minecraftforge.common.MinecraftForge;
 import net.minecraftforge.event.AddReloadListenerEvent;
 import net.minecraftforge.eventbus.api.IEventBus;
@@ -63,7 +65,7 @@ class SideProxy implements ISideProxy {
 	}
 	
 	@Override
-	public ClientLevel getClientWorld() {
+	public Level getClientWorld() {
 		return null;
 	}
 	
@@ -79,12 +81,14 @@ class SideProxy implements ISideProxy {
 		private static void postSetup(FMLLoadCompleteEvent event) {}
 		
 		@Override
+		@OnlyIn(Dist.CLIENT)
 		public Player getClientPlayer() {
 			return Minecraft.getInstance().player;
 		}
 		
 		@Override
-		public ClientLevel getClientWorld() {
+		@OnlyIn(Dist.CLIENT)
+		public Level getClientWorld() {
 			return Minecraft.getInstance().level;
 		}
 	}