Fix setup.py
More tests