Fix python issue