optimize code

Signed-off-by: Manson Li <manson.li@matchstick.tv>