Cosmetic fixed from previous PR

Signed-off-by: Marius Avram <marius.avram1309@gmail.com>
Merge pull request #533 from marius-avram/cosmetic_fixes2

Cosmetic fixed from previous PR