another quick fix for new UF api change