Adding for ps simulation. Note: Should turn back on the 2.2 shape.