Offset account overide using Context Costing for Elements

Hi Experts,

We have a requirement to modify the Offset Account Segment Values based on a custom logic at element level. Earlier for the same implementation we had modified the Cost Account Segments using context costing using Element Input values. Below was the approach.

First the Input values were created and in the reference attribute of the input value we selected the corresponding Costing segment(Eg COSTING_COMPANY).

Then further this Input value's direct result was created with the formula local variable in Status Processing Rule which was pulling the desired output based on the custom logic(value set).

However for the offset account we cannot use the same logic as in the Reference attribute of the Input Values only Costing Segments appear in the LOV not the Offset Segments.