Remove unnecessary import in tests

Just a little flake8 fix.