Fix the tests

Signed-off-by: Steven Hoffman <git@fustrate.com>