Added a test and fixed functional break that PEP8 changes made.