Move choice of contextual element into morph

Slight rearrangement of logic to simplify the template.