Fixed PEP8 issues with most recent PR.