"piston_extension" can be destroyed by TNT or TNT cart.

If a TNT explode while a piston is moving, the "piston_extension" can be destroyed and leave the "piston_base".

I think you can set the resistance of "piston_extension" to 6000000 to prevent it.