Cosmetic fixed from previous PR

Signed-off-by: Marius Avram <marius.avram1309@gmail.com>