diff --git a/build.gradle b/build.gradle
index 64b1d8b..6f398d0 100644
--- a/build.gradle
+++ b/build.gradle
@@ -15,7 +15,7 @@ apply plugin: 'net.minecraftforge.gradle'
 apply plugin: 'org.spongepowered.mixin'
 
 group = 'com.buuz135'
-version = '1.0.0'
+version = '1.18.1-1.0.1'
 
 java {
     archivesBaseName = 'DarkModeEverywhere'
diff --git a/src/main/java/com/buuz135/darkmodeeverywhere/DarkModeEverywhere.java b/src/main/java/com/buuz135/darkmodeeverywhere/DarkModeEverywhere.java
index 0b9a076..8522714 100644
--- a/src/main/java/com/buuz135/darkmodeeverywhere/DarkModeEverywhere.java
+++ b/src/main/java/com/buuz135/darkmodeeverywhere/DarkModeEverywhere.java
@@ -5,10 +5,10 @@
 import net.minecraftforge.fml.IExtensionPoint;
 import net.minecraftforge.fml.ModLoadingContext;
 import net.minecraftforge.fml.common.Mod;
+import net.minecraftforge.fml.loading.FMLEnvironment;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 
-// The value here should match an entry in the META-INF/mods.toml file
 @Mod("darkmodeeverywhere")
 public class DarkModeEverywhere {
 
@@ -17,7 +17,9 @@ public class DarkModeEverywhere {
 
     public DarkModeEverywhere() {
         ModLoadingContext.get().registerExtensionPoint(IExtensionPoint.DisplayTest.class, () -> new IExtensionPoint.DisplayTest(() -> "ANY", (remote, isServer) -> true));
-        DistExecutor.safeCallWhenOn(Dist.CLIENT, () -> ClientProxy::new);
+        if (FMLEnvironment.dist == Dist.CLIENT){
+            new ClientProxy();
+        }
     }
 
 }