Fix failing tests

The PEP8 tests were failing after my previous commit, so this fixes them up.