Fixing pep8 warning