Update setup.py

Change-Id: I816bd6ec3018fd50e0b153badc69a3bda7014341