Minor fixed (pep8)