Fixes the tests... hopefully

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