refactor

Signed-off-by: TimGuan <gkqwt1990@gmail.com>