fix pyflakes in test file