Fix python test