Fix connection unit tests.