Adjusted fix to adhere to PEP8.