fix? needs compared to paper behavior

src/main/java/io/github/rektroth/whiteout/mixin/entity/passive/BeeEntityMixin.java

@@ -0,0 +1,45 @@
+package io.github.rektroth.whiteout.mixin.entity.passive;
+
+import net.minecraft.entity.EntityType;
+import net.minecraft.entity.ai.control.FlightMoveControl;
+import net.minecraft.entity.ai.control.MoveControl;
+import net.minecraft.entity.mob.MobEntity;
+import net.minecraft.entity.passive.AnimalEntity;
+import net.minecraft.entity.passive.BeeEntity;
+import net.minecraft.world.World;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.Redirect;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+
+@Mixin(BeeEntity.class)
+public abstract class BeeEntityMixin extends AnimalEntity {
+    protected BeeEntityMixin(EntityType<? extends AnimalEntity> entityType, World world) {
+        super(entityType, world);
+    }
+
+    @Inject(
+        at = @At("TAIL"),
+        method = "<init>(Lnet/minecraft/entity/EntityType;Lnet/minecraft/world/World;)V"
+    )
+    public void fixedMoveControl(EntityType<? extends BeeEntity> entityType, World world, CallbackInfo ci) {
+        class BeeFlyingMoveControl extends FlightMoveControl {
+            public BeeFlyingMoveControl(final MobEntity entity, final int maxPitchChange, final boolean noGravity) {
+                super(entity, maxPitchChange, noGravity);
+            }
+
+            @Override
+            public void tick() {
+                if (!BeeEntityMixin.this.getWorld().isInBuildLimit(this.entity.getBlockPos())) {
+                    this.entity.setNoGravity(false);
+                }
+
+                super.tick();
+            }
+        }
+
+        this.moveControl = new BeeFlyingMoveControl(this, 20, true);
+    }
+}

src/main/resources/whiteout.mixins.json

@@ -9,6 +9,7 @@
 		"entity.ai.goal.WolfBegGoalMixin",
 		"entity.boss.dragon.EnderDragonFightAccessor",
 		"entity.boss.dragon.EnderDragonFightMixin",
+		"entity.passive.BeeEntityMixin",
 		"server.PlayerManagerAccessor",
 		"server.PlayerManagerMixin",
 		"server.dedicated.command.DeOpCommandMixin",