Fix python error