Forgot to update setup.py