PEP8 E401 fix