Fixing PEP8

Not sure why the behavior is different on travis and on my machine but it
gets a bit irritating sometimes