Pistons do not retract with the block infront of it in y-level 2, very annoying for creating builds low in the ground in superflats

How to reproduce:

Be in BE version 1.16.201 hotfix

Make a superflat world

Dig into layer 2 (just above bedrock) 

Put a sticky piston, put a block ontop of it, extend it, and retract it

Expected result:

The block will not retract with the piston.