Small improvement in setup.py to build easier