Another fix in setup.py

Signed-off-by: Bilal Aslam <maslam@gmail.com>