fix missed refactor