Updated syntax to pass flake8