remove unused import to fix flake8 error

Signed-off-by: David Caplan <dcaplan@gmail.com>