diff --git a/src/main/java/net/mcreator/target/mobeffect/ShockMobEffect.java b/src/main/java/net/mcreator/target/mobeffect/ShockMobEffect.java index c52f23ec6..41dd55090 100644 --- a/src/main/java/net/mcreator/target/mobeffect/ShockMobEffect.java +++ b/src/main/java/net/mcreator/target/mobeffect/ShockMobEffect.java @@ -54,22 +54,17 @@ public class ShockMobEffect extends MobEffect { @SubscribeEvent public static void onEffectAdded(MobEffectEvent.Added event) { LivingEntity living = event.getEntity(); + + MobEffectInstance instance = event.getEffectInstance(); + if (!instance.getEffect().equals(TargetModMobEffects.SHOCK.get())) { + return; + } + + living.hurt(TargetModDamageTypes.causeShockDamage(living.level().registryAccess(), + event.getEffectSource()), 5.0f); + if (event.getEffectSource() instanceof LivingEntity source) { - MobEffectInstance instance = event.getEffectInstance(); - - if (!instance.getEffect().equals(TargetModMobEffects.SHOCK.get())) { - return; - } - living.getPersistentData().putInt("TargetShockAttacker", source.getId()); - - if (living.hasEffect(TargetModMobEffects.SHOCK.get())) { - System.out.println(instance.getDuration()); - if (instance.getDuration() % 10 == 0) { - living.hurt(TargetModDamageTypes.causeShockDamage(living.level().registryAccess(), - source), 5.0f); - } - } } }