Minor code refactoring

Signed-off-by: Oleg Nenashev <o.v.nenashev@gmail.com>