Fix previously introduced bugs

Signed-off-by: emanuele <emanuele45@gmail.com>