Fix python module