Missed refactor