small refactoring

Signed-off-by: pussbb <pussbb@gmail.com>