Fix setup.py to install the correct package

Change-Id: I87e3977ecd9078df0c8aa14ed18dc288fe49c596