Fix setup.py

Signed-off-by: Olivier Mehani <olivier.mehani@nicta.com.au>