When checking for air above it, I will get stuck on top of a block

I created a block that will get struck by lightning randomly when it's in a certain biome. It checks for its current biome as well as if there's air above it. When I stand on it, I'll get stuck unless I change my Y position, but not X or Z, as is displayed here: https://imgur.com/a/UiLKCda (sorry for the bad cropping and watermark)

This is the code used

Does anyone know how to fix this while also keeping the lightning?