fix python test file