Fixed pep8-ness of fix for #18