Piston not retracting

Pistons don't retract in some situations in the Z axis.

 

Here is a video of me showing the bug: https://www.youtube.com/watch?v=lbzP3o0kwAM

The bug also works in different axis. But it looks like a redstone block two blocks above the piston will keep powering the piston for some reason and once that block is removed the piston does not get updated. Here are two more clips:

https://youtu.be/yZHwPIByoPg

https://youtu.be/j_KZn5Gu4EE