I missed this in the refactor