Fixing tests

Signed-off-by: Jason Myers <jason@jasonamyers.com>