Forgotten changes related with the previous commit

Signed-off-by: Kitiara <vinas_solamnus1@hotmail.com>