Hi @Matthew.Latham,
You forgot to mention, but i asume you are using the Python code of @Daniel_Woodcock1 from the topic below.
Have you also seen the bottom comment ?
Changing that line of code made it work for me (on multiple elements).
Kind regards,
Mark