From 51345a1cc84f21af20c80a79906905d73ba1b385 Mon Sep 17 00:00:00 2001
From: Bjarne Koll <git@lynxplay.dev>
Date: Fri, 2 May 2025 18:57:18 +0200
Subject: [PATCH] Correct nullable fall location type

---
 .../paper/world/damagesource/PaperCombatTrackerWrapper.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/paper-server/src/main/java/io/papermc/paper/world/damagesource/PaperCombatTrackerWrapper.java b/paper-server/src/main/java/io/papermc/paper/world/damagesource/PaperCombatTrackerWrapper.java
index 323f97ea4..43afa79e9 100644
--- a/paper-server/src/main/java/io/papermc/paper/world/damagesource/PaperCombatTrackerWrapper.java
+++ b/paper-server/src/main/java/io/papermc/paper/world/damagesource/PaperCombatTrackerWrapper.java
@@ -6,6 +6,7 @@ import io.papermc.paper.adventure.PaperAdventure;
 import java.util.ArrayList;
 import java.util.List;
 import net.kyori.adventure.text.Component;
+import net.minecraft.Optionull;
 import net.minecraft.Util;
 import net.minecraft.world.damagesource.FallLocation;
 import org.bukkit.entity.LivingEntity;
@@ -75,7 +76,7 @@ public record PaperCombatTrackerWrapper(
     @Override
     public FallLocationType calculateFallLocationType() {
         final FallLocation fallLocation = FallLocation.getCurrentFallLocation(this.handle().mob);
-        return fallLocation == null ? FallLocationType.GENERIC : PaperCombatTrackerWrapper.minecraftToPaper(fallLocation);
+        return Optionull.map(fallLocation, PaperCombatTrackerWrapper::minecraftToPaper);
     }
 
     private static final BiMap<FallLocation, FallLocationType> FALL_LOCATION_MAPPING = Util.make(() -> {