Fix pep8 comment issue