Fixed Python style issue in tests.