--- a/net/minecraft/util/datafix/fixes/ChunkBedBlockEntityInjecterFix.java
+++ b/net/minecraft/util/datafix/fixes/ChunkBedBlockEntityInjecterFix.java
@@ -42,8 +42,8 @@
         return TypeRewriteRule.seq(
             this.fixTypeEverywhere(
                 "InjectBedBlockEntityType",
-                this.getInputSchema().findChoiceType(References.BLOCK_ENTITY),
-                this.getOutputSchema().findChoiceType(References.BLOCK_ENTITY),
+                (com.mojang.datafixers.types.templates.TaggedChoice.TaggedChoiceType<String>)this.getInputSchema().findChoiceType(References.BLOCK_ENTITY),
+                (com.mojang.datafixers.types.templates.TaggedChoice.TaggedChoiceType<String>)this.getOutputSchema().findChoiceType(References.BLOCK_ENTITY),
                 p_184841_ -> p_184837_ -> p_184837_
             ),
             this.fixTypeEverywhereTyped(
