Missed one spot for main refactor