fixes to python test