1 files changed, 6 insertions, 1 deletions
diff --git a/src/main/java/world/anhgelus/lifesteal/LifeStealer.java b/src/main/java/world/anhgelus/lifesteal/LifeStealer.java
index 2d1268a..409e3d6 100644
--- a/src/main/java/world/anhgelus/lifesteal/LifeStealer.java
+++ b/src/main/java/world/anhgelus/lifesteal/LifeStealer.java
@@ -60,9 +60,14 @@ public class LifeStealer {
     }
 
     public ItemStack getHeart() {
+        return turnsIntoHeart(new ItemStack(Items.NETHER_STAR));
+    }
+
+    public ItemStack turnsIntoHeart(ItemStack is) {
+        if (!is.isOf(Items.NETHER_STAR))
+            throw new IllegalArgumentException("Cannot convert a stack into an heart if it's not a nether star");
         final var nbt = new NbtCompound();
         nbt.putString(PLAYER_KEY, player.getUuid().toString());
-        final var is = new ItemStack(Items.NETHER_STAR);
         is.set(DataComponentTypes.CUSTOM_NAME, Text.empty().append(Text.of("Heart of ")).append(player.getName()));
         is.set(DataComponentTypes.CUSTOM_DATA, NbtComponent.of(nbt));
         return is;