diff --git a/arclight-common/src/main/java/io/izzel/arclight/common/mixin/bukkit/CraftEventFactoryMixin.java b/arclight-common/src/main/java/io/izzel/arclight/common/mixin/bukkit/CraftEventFactoryMixin.java
index 03416a2a..c0a8d23b 100644
--- a/arclight-common/src/main/java/io/izzel/arclight/common/mixin/bukkit/CraftEventFactoryMixin.java
+++ b/arclight-common/src/main/java/io/izzel/arclight/common/mixin/bukkit/CraftEventFactoryMixin.java
@@ -29,7 +29,7 @@ public class CraftEventFactoryMixin {
         BlockPos damageEventBlock = ArclightCaptures.getDamageEventBlock();
         if (damageEventEntity != null && entityDamage == null) {
             if (source.damageType.equals(DamageSource.LIGHTNING_BOLT.damageType)) {
-                entityDamage = entity;
+                entityDamage = damageEventEntity;
             }
         }
         if (damageEventBlock != null && blockDamage == null) {