Simplified code

Signed-off-by: servanter <fengshang@126.com>