[DD-854] code clean-up