Fix python setup script
Remove unnecessary module