Fixed python test