Small style fix.
Update to comply with pep8.