diff --git a/Masks/zora_mask.asm b/Masks/zora_mask.asm
index 7b1ba2d..7a6c1e0 100644
--- a/Masks/zora_mask.asm
+++ b/Masks/zora_mask.asm
@@ -205,7 +205,8 @@ pullpc
 .dungeon_resurface
 {
   LDA $1B : BEQ .return_default ; We are in overworld actually 
-
+  ; Check if we are swimming 
+  LDA $5D : CMP #$04 : BNE .return_default
   ; Check if the player is actually diving 
   LDA !ZoraDiving : BEQ .return_default