JZ: Fix flake8 fail