more fixes to python tests