General test improvement

Signed-off-by: Ken Barber <ken@bob.sh>