Fix unit tests

Signed-off-by: nigel <nigel.cook@hp.com>