Add protection against pistons