Updating setup.py to fix installation error.