Fix broken tests

Signed-off-by: Rick Waldron <waldron.rick@gmail.com>