Update tests for new connection pool