Clean up test_migrate. flask8 fix