Fix a crash

Author: Patbox

gradle.properties

@@ -9,7 +9,7 @@ org.gradle.parallel=true
 	loader_version=0.15.11
 
 # Mod Properties
-	mod_version = 1.0.0
+	mod_version = 1.0.1
 	maven_group = eu.pb4
 	archives_base_name = ouch
 

src/main/java/eu/pb4/ouch/Config.java

@@ -24,11 +24,10 @@ public static Config get() {
             DeathDisplayLogic.CODEC.listOf().listOf().fieldOf("death").forGetter(Config::death)
     ).apply(instance, Config::new));
 
-    @Nullable
     public void selectDamage(LivingEntity entity, DamageSource source, float amount, BiConsumer<Text, FloatingText.DisplaySettings> consumer) {
         var predicateContext = PredicateContext.of(entity);
         var attackerContext = source.getAttacker() != null ? PredicateContext.of(source.getAttacker()) : PredicateContext.of(Objects.requireNonNull(entity.getServer()));
-        var sourceContext = source.getSource() != null ? PredicateContext.of(source.getAttacker()) : PredicateContext.of(Objects.requireNonNull(entity.getServer()));
+        var sourceContext = source.getSource() != null ? PredicateContext.of(source.getSource()) : PredicateContext.of(Objects.requireNonNull(entity.getServer()));
         for (var a : this.damage) {
             for (var e : a) {
                 if (e.match(entity, source, predicateContext, attackerContext, sourceContext)) {
@@ -39,11 +38,10 @@ public void selectDamage(LivingEntity entity, DamageSource source, float amount,
         }
     }
 
-    @Nullable
     public void selectDeath(LivingEntity entity, DamageSource source, BiConsumer<Text, FloatingText.DisplaySettings> consumer) {
         var predicateContext = PredicateContext.of(entity);
         var attackerContext = source.getAttacker() != null ? PredicateContext.of(source.getAttacker()) : PredicateContext.of(Objects.requireNonNull(entity.getServer()));
-        var sourceContext = source.getSource() != null ? PredicateContext.of(source.getAttacker()) : PredicateContext.of(Objects.requireNonNull(entity.getServer()));
+        var sourceContext = source.getSource() != null ? PredicateContext.of(source.getSource()) : PredicateContext.of(Objects.requireNonNull(entity.getServer()));
         for (var a : this.death) {
             for (var e : a) {
                 if (e.match(entity, source, predicateContext, attackerContext, sourceContext)) {
@@ -55,7 +53,6 @@ public void selectDeath(LivingEntity entity, DamageSource source, BiConsumer<Tex
 
     }
 
-    @Nullable
     public void selectHealing(LivingEntity entity, float amount, BiConsumer<Text, FloatingText.DisplaySettings> consumer) {
         var predicateContext = PredicateContext.of(entity);
         for (var a : this.healing) {