Fixed error in python test.