You will need to include a policy to install the same Jamf Connect package you have your PreStage as an EnrollmentComplete trigger (or, if you have an existing script that runs at enrollment, include the Jamf Connect install policy in that script.)
This is what I do. For ADE enrollments, I have my Jamf Connect "Mega" package in my PreStage so it gets installed during enrollment. I also have a call in my enrollment script that installs the same package during my enrollment script. This ensures that it gets installed after a UIE enrollment. For ADE enrollments is just installs over the existing version. Doesn't see to affect anything.
