Help with Zelda Movable Blocks by NOPEname

I found this datapack that makes movable block by a youtuber named NOPEname, who made the Zelda Project, and I wanted to try using the data pack to work on something akin to the Strength Boulder Puzzles in Pokémon. My issue is that while the block appears to have some form of gravity, it is unable to sink into water or lava. Instead, it floats over the lava when it falls. I was wondering if there was some way to edit it so that the blocks can actually fall into lava or water to be able to act as a bridge?
Video Example of the Datapack: https://www.youtube.com/watch?v=RgjUX-3rKE4