Fix comment as per pep8