How to make a potion effect enable the entity to walk on powder snow?

I just managed to make a potion effect make the entity immune to freezing but the entity/player still can't walk on top of powder snow or jump out of it. I found a forum with a reply which solves it but only for armour pieces (the solution only works on wearable armour), it involved adding an @override in the code editor to turn canWalkOnPowderSnow to true and included "(itemStack stack, LivingEntity entity)", I simply removed "itemStack stack" but obviously that didn't compile, how do I change it so that it works on potion effects?