Improve tests

Signed-off-by: Eli Skeggs <eskeggs@globesherpa.com>