And fix the tests

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