[Silviu, Radu] refactor