fix piston bug