Pep8 fix

Fixes travis failing build