Flake8 mistake