Fix Pep8 issues with model tests