Update enum cycling list controller

1 files changed, 5 insertions, 9 deletions

diff --git a/src/main/java/me/xmrvizzy/skyblocker/config/ConfigUtils.java b/src/main/java/me/xmrvizzy/skyblocker/config/ConfigUtils.java
index f5605342..4a615ecc 100644
--- a/src/main/java/me/xmrvizzy/skyblocker/config/ConfigUtils.java
+++ b/src/main/java/me/xmrvizzy/skyblocker/config/ConfigUtils.java
@@ -1,15 +1,11 @@
 package me.xmrvizzy.skyblocker.config;
 
 import dev.isxander.yacl3.api.Option;
-import dev.isxander.yacl3.api.controller.CyclingListControllerBuilder;
-import net.minecraft.text.Text;
+import dev.isxander.yacl3.api.controller.EnumControllerBuilder;
 
 public class ConfigUtils {
-
-	@SuppressWarnings("unchecked")
-	public static <E extends Enum<?>> CyclingListControllerBuilder<E> createCyclingListController4Enum(Option<E> opt) {
-		E[] constants = (E[]) opt.binding().defaultValue().getClass().getEnumConstants();
-		
-		return CyclingListControllerBuilder.create(opt).values(constants).formatValue(formatter -> Text.of(formatter.toString()));
-	}
+    @SuppressWarnings("unchecked")
+    public static <E extends Enum<E>> EnumControllerBuilder<E> createEnumCyclingListController(Option<E> opt) {
+        return EnumControllerBuilder.create(opt).enumClass((Class<E>) opt.binding().defaultValue().getClass());
+    }
 }