Camunda 8 - Force DMN result value to be Context

Hello,
I’m using the DmnEngine in a Java project. In our use case, we use DMNs with a dynamic amount of output columns (mostly between 1-3). When having 2 or 3 output columns in the rule sheet, we are given a ValContext which is then mapped into a POJO on java field name = DMN output column name. However, when using only one column, we are given a primitive return type (e.g. number), so we are unable to map using column name.

Is there any way to force the return value of DMNs with one column to return contexts? Or another way to get access to the name of the output column to use in our rule value mapper?

Right now, we use dummy columns for DMNs with one column, which we would like to avoid.
Thank you!