Remove unused import

Signed-off-by: walteryang47 <walteryang47@gmail.com>