Update JEXT support in set_functions