--- a/net/minecraft/util/datafix/fixes/AddNewChoices.java
+++ b/net/minecraft/util/datafix/fixes/AddNewChoices.java
@@ -30,8 +30,8 @@
         if (p_14639_.getKeyType() != p_14640_.getKeyType()) {
             throw new IllegalStateException("Could not inject: key type is not the same");
         } else {
-            return this.fixTypeEverywhere(p_14638_, p_14639_, p_14640_, p_14636_ -> p_145061_ -> {
-                    if (!p_14640_.hasType(p_145061_.getFirst())) {
+            return this.fixTypeEverywhere(p_14638_, p_14639_, (TaggedChoiceType<K>)p_14640_, p_14636_ -> p_145061_ -> {
+                    if (!((TaggedChoiceType<K>)p_14640_).hasType(p_145061_.getFirst())) {
                         throw new IllegalArgumentException(String.format(Locale.ROOT, "Unknown type %s in %s ", p_145061_.getFirst(), this.type));
                     } else {
                         return p_145061_;
