Fix up setup.py

Signed-off-by: Amr Ali <amr@databracket.com>