Fix for Flake8 error