From 364ddf7c6de054f358afd67a1507f1a26b0ec94d Mon Sep 17 00:00:00 2001
From: PepperCode1 <44146161+PepperCode1@users.noreply.github.com>
Date: Tue, 7 Mar 2023 14:46:08 -0800
Subject: [PATCH] Fix #4454

---
 .../create/content/curiosities/armor/DivingHelmetItem.java      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/com/simibubi/create/content/curiosities/armor/DivingHelmetItem.java b/src/main/java/com/simibubi/create/content/curiosities/armor/DivingHelmetItem.java
index a9a3b346d..2dbfb2e1a 100644
--- a/src/main/java/com/simibubi/create/content/curiosities/armor/DivingHelmetItem.java
+++ b/src/main/java/com/simibubi/create/content/curiosities/armor/DivingHelmetItem.java
@@ -39,7 +39,7 @@ public class DivingHelmetItem extends CopperArmorItem {
 			return;
 
 		boolean lavaDiving = entity.isEyeInFluidType(ForgeMod.LAVA_TYPE.get());
-		if (!entity.isEyeInFluidType(ForgeMod.WATER_TYPE.get()) && !lavaDiving)
+		if (!entity.canDrownInFluidType(entity.getEyeInFluidType()) && !lavaDiving)
 			return;
 		if (entity instanceof Player && ((Player) entity).isCreative())
 			return;