diff --git a/forge/src/main/java/dev/engine_room/flywheel/impl/FlywheelForge.java b/forge/src/main/java/dev/engine_room/flywheel/impl/FlywheelForge.java
index 7bef593f8..2bc2f9cd9 100644
--- a/forge/src/main/java/dev/engine_room/flywheel/impl/FlywheelForge.java
+++ b/forge/src/main/java/dev/engine_room/flywheel/impl/FlywheelForge.java
@@ -101,7 +101,7 @@ public final class FlywheelForge {
 			if (e.getRegistryKey().equals(Registries.COMMAND_ARGUMENT_TYPE)) {
 				e.register(Registries.COMMAND_ARGUMENT_TYPE, Flywheel.rl("backend"), () -> BackendArgument.INFO);
 				e.register(Registries.COMMAND_ARGUMENT_TYPE, Flywheel.rl("debug_mode"), () -> DebugModeArgument.INFO);
-				e.register(ForgeRegistries.Keys.COMMAND_ARGUMENT_TYPES, Flywheel.rl("light_smoothness"), () -> LightSmoothnessArgument.INFO);
+				e.register(Registries.COMMAND_ARGUMENT_TYPE, Flywheel.rl("light_smoothness"), () -> LightSmoothnessArgument.INFO);
 			}
 		});
 	}
diff --git a/forge/src/main/java/dev/engine_room/flywheel/impl/ForgeFlwConfig.java b/forge/src/main/java/dev/engine_room/flywheel/impl/ForgeFlwConfig.java
index 9e897ff3b..db8563519 100644
--- a/forge/src/main/java/dev/engine_room/flywheel/impl/ForgeFlwConfig.java
+++ b/forge/src/main/java/dev/engine_room/flywheel/impl/ForgeFlwConfig.java
@@ -98,9 +98,9 @@ public class ForgeFlwConfig implements FlwConfig {
 	}
 
 	public static class ForgeBackendConfig implements BackendConfig {
-		public final ForgeConfigSpec.EnumValue<LightSmoothness> lightSmoothness;
+		public final ModConfigSpec.EnumValue<LightSmoothness> lightSmoothness;
 
-		public ForgeBackendConfig(ForgeConfigSpec.Builder builder) {
+		public ForgeBackendConfig(ModConfigSpec.Builder builder) {
 			lightSmoothness = builder.comment("How smooth flywheel's shader-based lighting should be. May have a large performance impact.")
 					.defineEnum("lightSmoothness", LightSmoothness.SMOOTH);
 		}