Auto Material Tag not working when integrated into full script

Hi, hopefully some bright minds can take 5 mins to look over my script & file, I’m having trouble getting the material tags to behave.

The script creates small sections of each internal wall type in an ** Admin ** phase, creates callouts centred on these, adds a dimension and tags each of the layer materials.

When testing parts of the script, I’ve got the material tag workflow to behave as intended, by selecting walls using a Select Model Elements node. However, once integrated into my full script, they seem to lose their host.

From a suggestion from @jacob.small, I followed this workflow (from @Alban_de_Chasteigner) to create & position the tags:

For anyone who does take the time to look into this for me, if you head to the ** ADMIN ** floorplan before running the script, it should allow you easy navigation of the created callouts.

Edit: Revit 2023.1, Dynamo 2.16