Fix failing tests

Signed-off-by: Fred Bricon <fbricon@gmail.com>