refactor: refine argument type