flake8 fixes
Fixed variable name