Minor fix of setup.py.