Fixed setup

Fixed setup.py