code review

Signed-off-by: Artsiom Marozau <Artsiom_Marozau@epam.com>