small fix from refactor