Solved error shown in flake8