Fixing failing python test.