Fixes to tests

Signed-off-by: Nick Brook <nick.brook@ratemyplacement.co.uk>