Fix dist package run unit test failed.

Signed-off-by: Wei Luo <luowei@yahoo-inc.com>