missed one part of refactor