Piston shaft assembly upgrade