forgot one small refactor