refactor - fix warning