Control Rig Module offset not working correctly

Working with the new Control Rig Modules of UE 5.4.

I’m having problems with the shape offset when trying to position the control shape outside the character. It’s a custom module based on the “AddControl” rig module but the original behaves the same.
This one:
UE54_AddControl

What happens is when setting a shape offset value the transforms won’t work correctly anymore.
Tried different approaches like passing the existing “Control Stack at Item” function a shape offset:


As well as calling “Set Control Offset” for the offset afterwards but it’s always the same issue.

It’s not from my adjustments as the original “AddControl” rig module behaves the same.
Results with the original module on a leg are following.
Without offset it works as expected:

With offset using same transforms it breaks:

Anyone knows how to adjust the “AddControl” module to have it work correctly with shape offsets?