Added prep for runs with varying axial shield height with endcap