Simple refactoring.

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