1 files changed, 6 insertions, 0 deletions
diff --git a/src/main/java/world/anhgelus/lifesteal/LifeSteal.java b/src/main/java/world/anhgelus/lifesteal/LifeSteal.java
index 3cd9247..8ebf9d6 100644
--- a/src/main/java/world/anhgelus/lifesteal/LifeSteal.java
+++ b/src/main/java/world/anhgelus/lifesteal/LifeSteal.java
@@ -88,6 +88,7 @@ public class LifeSteal implements ModInitializer {
         ServerLivingEntityEvents.AFTER_DEATH.register((LivingEntity entity, DamageSource damageSource) -> {
             if (!(entity instanceof ServerPlayerEntity player)) return;
             getLifeStealer(player).death();
+            save(player);
             if (damageSource.getAttacker() instanceof ServerPlayerEntity killer) getLifeStealer(killer).kill();
         });
 
@@ -112,7 +113,12 @@ public class LifeSteal implements ModInitializer {
             // if the heart was not added
             if (!getLifeStealer(serverPlayer).kill()) return ActionResult.PASS;
             player.getMainHandStack().decrement(1);
+            save(serverPlayer);
             return ActionResult.SUCCESS;
         });
     }
+
+    private void save(ServerPlayerEntity player) {
+        LifeStealer.Manager.savePlayers(StateSaver.getServerState(player.getEntityWorld().getServer()));
+    }
 }