Fix the test failure.
PEP 8 cleanup.