Fix for flake8 checks on new test file.