code refactor
fix warning