diff --git a/lib/xos-genx/xosgenx/jinja2_extensions/tosca.py b/lib/xos-genx/xosgenx/jinja2_extensions/tosca.py
index 9e3b4a8..996d63d 100644
--- a/lib/xos-genx/xosgenx/jinja2_extensions/tosca.py
+++ b/lib/xos-genx/xosgenx/jinja2_extensions/tosca.py
@@ -33,11 +33,11 @@
     else:
         return type
 
-def xproto_fields_to_tosca_keys(fields):
+def xproto_fields_to_tosca_keys(fields, m):
     keys = []
 
     # look for one_of keys
-    _one_of = xproto_field_graph_components(fields, 'tosca_key_one_of')
+    _one_of = xproto_field_graph_components(fields, m, 'tosca_key_one_of')
     one_of = [list(i) for i in _one_of]
 
     # look for explicit keys