Fixing setup.py for distribution.

Change-Id: I3fc73f403158a39f32613db3fb0e12fabfde8158