diff --git a/src/main/java/dev/micle/firefly_bush_backport/block/FireflyBushBlock.java b/src/main/java/dev/micle/firefly_bush_backport/block/FireflyBushBlock.java
index 4ba0ded..25415a8 100644
--- a/src/main/java/dev/micle/firefly_bush_backport/block/FireflyBushBlock.java
+++ b/src/main/java/dev/micle/firefly_bush_backport/block/FireflyBushBlock.java
@@ -1,5 +1,6 @@
 package dev.micle.firefly_bush_backport.block;
 
+import com.mojang.serialization.MapCodec;
 import dev.micle.firefly_bush_backport.config.Config;
 import dev.micle.firefly_bush_backport.particle.ModParticles;
 import dev.micle.firefly_bush_backport.sound.ModSounds;
@@ -14,16 +15,25 @@ import net.minecraft.world.level.block.BonemealableBlock;
 import net.minecraft.world.level.block.BushBlock;
 import net.minecraft.world.level.block.state.BlockState;
 import net.minecraft.world.level.levelgen.Heightmap;
+import org.jetbrains.annotations.NotNull;
 
 import javax.annotation.ParametersAreNonnullByDefault;
 import java.util.List;
 import java.util.Optional;
 
 public class FireflyBushBlock extends BushBlock implements BonemealableBlock {
+    public static final MapCodec<FireflyBushBlock> CODEC = simpleCodec(FireflyBushBlock::new);
+
     public FireflyBushBlock(Properties properties) {
         super(properties);
     }
 
+    @Override
+    @NotNull
+    protected MapCodec<? extends BushBlock> codec() {
+        return CODEC;
+    }
+
     @Override
     @ParametersAreNonnullByDefault
     public void animateTick(BlockState blockState, Level level, BlockPos blockPos, RandomSource randomSource) {
@@ -58,7 +68,7 @@ public class FireflyBushBlock extends BushBlock implements BonemealableBlock {
 
     @Override
     @ParametersAreNonnullByDefault
-    public boolean isValidBonemealTarget(LevelReader level, BlockPos blockPos, BlockState blockState, boolean isClient) {
+    public boolean isValidBonemealTarget(LevelReader level, BlockPos blockPos, BlockState blockState) {
         return hasSpreadableNeighbourPos(level, blockPos, blockState);
     }