Fix some flake8 lint error.

Just remove some unused import