Preliminary python test fix