Clean up tests

Signed-off-by: Steven Downer <grinch@outlook.com>