a little bit clean up.

Signed-off-by: Shanyu Zhao <shzhao@microsoft.com>