Fix tests

Signed-off-by: Derek McGowan <derek@mcgstyle.net>