Fix tests broken by #21

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