--- a/net/minecraft/world/gen/treedecorator/TreeDecoratorType.java
+++ b/net/minecraft/world/gen/treedecorator/TreeDecoratorType.java
@@ -3,7 +3,7 @@
 import com.mojang.serialization.Codec;
 import net.minecraft.util.registry.Registry;
 
-public class TreeDecoratorType<P extends TreeDecorator> {
+public class TreeDecoratorType<P extends TreeDecorator> extends net.minecraftforge.registries.ForgeRegistryEntry<TreeDecoratorType<?>> {
    public static final TreeDecoratorType<TrunkVineTreeDecorator> field_227425_a_ = func_236877_a_("trunk_vine", TrunkVineTreeDecorator.field_236878_a_);
    public static final TreeDecoratorType<LeaveVineTreeDecorator> field_227426_b_ = func_236877_a_("leave_vine", LeaveVineTreeDecorator.field_236870_a_);
    public static final TreeDecoratorType<CocoaTreeDecorator> field_227427_c_ = func_236877_a_("cocoa", CocoaTreeDecorator.field_236866_a_);
