code enhance.

Signed-off-by: August Yip <augustyip@ecvision.com>