[Refator]Optimize import clause

Signed-off-by: Ethan Shan <shan0xiao0xi@gmail.com>