tiny refactor fixing error