Entity of block pushed/pulled by piston changes shape/glitches out

When a sticky piston is extending or retracting a block, the entity it turns into is almost 3 blocks wide when the piston retracts into or extends from the floor.

When a sticky piston pushes or pulls above the floor, it only glitches and loses some faces.