Fix unused import

Signed-off-by: eggfly <lihaohua90@gmail.com>