setup.py fix.

Just a small change...