Fix the tests.
pyflakes