Add some connection tests and fix build failing