update tests to include the DB