Update unit tests to fix issue with database pool.