Fixed incompatibility with flake8