Update tests to Flake8 message changes