How to fix a materials orientation on an object?

Hi, I’m attempting to answer my own question from this here. What I’ve done is take the material from the VR Template that outlines the teleportation ring and applied it to a cylinder that I will attach to my engine. However, given the blueprint from the VR template, it uses the world position and object orientation to always orient the falloff upwards - no matter what the rotation of the object is. I’ve tried several different fixes but none produce the same result. I’ve attached the BP for the original material MatBP01.

Is there an easy alternative to the nodes in the bottom left that will align the material with the fixed z axis of the object it is applied to?

Thank you in advance for your help.

I’ve attached what it does currently and what I want it to look like in all orientations as well.