Cleaned up tests with pyflakes.