--- a/net/minecraft/util/datafix/fixes/BlockEntityIdFix.java
+++ b/net/minecraft/util/datafix/fixes/BlockEntityIdFix.java
@@ -47,8 +47,8 @@
     public TypeRewriteRule makeRule() {
         Type<?> type = this.getInputSchema().getType(References.ITEM_STACK);
         Type<?> type1 = this.getOutputSchema().getType(References.ITEM_STACK);
-        TaggedChoiceType<String> taggedchoicetype = this.getInputSchema().findChoiceType(References.BLOCK_ENTITY);
-        TaggedChoiceType<String> taggedchoicetype1 = this.getOutputSchema().findChoiceType(References.BLOCK_ENTITY);
+        TaggedChoiceType<String> taggedchoicetype = (TaggedChoiceType<String>)this.getInputSchema().findChoiceType(References.BLOCK_ENTITY);
+        TaggedChoiceType<String> taggedchoicetype1 = (TaggedChoiceType<String>)this.getOutputSchema().findChoiceType(References.BLOCK_ENTITY);
         return TypeRewriteRule.seq(
             this.convertUnchecked("item stack block entity name hook converter", type, type1),
             this.fixTypeEverywhere(
