Update for latest piston