Fixing tests

Signed-off-by: Alan Gardner <alan@armoin.com>