update for new piston changes