Fixed a failing python test.

Reviewed by Csaba Osztrogonác.