PEP8 cosmetics
Fixed bug