Fixed the setup.py script to include the correct package.