diff --git a/gm4_end_fishing/data/gm4_end_fishing/function/enderpuff/process.mcfunction b/gm4_end_fishing/data/gm4_end_fishing/function/enderpuff/process.mcfunction
index 194b8a0231..92df86e42c 100644
--- a/gm4_end_fishing/data/gm4_end_fishing/function/enderpuff/process.mcfunction
+++ b/gm4_end_fishing/data/gm4_end_fishing/function/enderpuff/process.mcfunction
@@ -6,6 +6,6 @@ tag @s remove gm4_enderpuff_equipped
 tag @s[predicate=gm4_end_fishing:holding_enderpuff] add gm4_enderpuff_equipped
 
 # apply levitation
-execute unless block ~ ~-1 ~ #gm4:no_collision run tag @s remove gm4_enderpuff_used
+execute if predicate {condition:"minecraft:entity_properties",entity:"this",predicate:{flags:{is_on_ground:1b}}} run tag @s remove gm4_enderpuff_used
 execute if entity @s[tag=gm4_enderpuff_equipped,tag=!gm4_enderpuff_used] run function gm4_end_fishing:enderpuff/apply_levitation
 execute if entity @s[tag=gm4_enderpuff_levitation,tag=!gm4_enderpuff_equipped] run function gm4_end_fishing:enderpuff/clear_levitation