Fix failing test

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