Forgot some modifications

Signed-off-by: Radu Ungureanu <radu.ungureanu1912@cti.pub.ro>