Extended Piston - Redstoneblock Bug

I found a bug in the game, which occurs when I power a piston with redstone block on top of it. The piston extends but if I put the power away from the piston it is still extended.
So the problem ist that a normal piston and a sticky piston keeps being extended if a redstone block is on top of it and it isnt powered anymore.