diff --git a/src/main/java/draylar/gofish/mixin/FishingRodPropertyMixin.java b/src/main/java/draylar/gofish/mixin/FishingRodPropertyMixin.java index c3bfad2..2a5e942 100644 --- a/src/main/java/draylar/gofish/mixin/FishingRodPropertyMixin.java +++ b/src/main/java/draylar/gofish/mixin/FishingRodPropertyMixin.java @@ -20,7 +20,7 @@ import org.spongepowered.asm.mixin.Unique; 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.ModifyArg; import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable; import java.util.ArrayList; @@ -38,13 +38,13 @@ private void storeContext(World world, PlayerEntity user, Hand hand, CallbackInf this.player = user; } - @Redirect(method = "use", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/World;spawnEntity(Lnet/minecraft/entity/Entity;)Z")) - private boolean modifyBobber(World world, Entity entity) { + @ModifyArg(method = "use", at = @At(value = "INVOKE", target = "Lnet/minecraft/world/World;spawnEntity(Lnet/minecraft/entity/Entity;)Z")) + private Entity modifyBobber(Entity entity) { + World world = entity.getWorld(); if(entity instanceof FishingBobberEntity bobber) { modifyBobber(world, bobber); } - - return world.spawnEntity(entity); + return entity; } @Unique