simplify code

Signed-off-by: xbz <xbz.o3sig@gmail.com>