Fix up the test suite

Signed-off-by: Graham Campbell <graham@mineuk.com>