optimize code

Signed-off-by: Manson Li <lijinsong@intfhink.com>